Eka Robotics
Machine Learning / Reinforcement Learning Infrastructure Engineer
Boston Area · On-site · Full-time · Posted Mar 8, 2025
- Education
- BS, MS or higher in Computer Science, Computer Engineering, Machine Learning or a related technical field.
About the job
Build and maintain large-scale training infrastructure and developer tools for robotics reinforcement and machine learning, supporting research from prototyping through distributed production runs.
What you'll do
- Design and operate training systems for orchestration, scheduling, checkpointing, and experiment tracking.
- Build tools for launching, monitoring, debugging, and reproducing experiments, and scale training pipelines across compute clusters.
- Manage cloud compute resources and collaborate with researchers on training practices and core JAX model and training code.
What they ask for
- BS, MS or higher in Computer Science, Computer Engineering, Machine Learning or a related technical field.
- Strong software engineering fundamentals and a proven track record building ML training infrastructure, internal developer platforms, or scalable systems.
- Hands-on large-scale training experience with JAX, PyTorch, or TensorFlow.
- Experience or familiarity with distributed training, multi-host setups, data pipelines, cloud platforms or orchestration systems, automated testing, CI/CD for ML workflows, and logging/telemetry.
Nice to have
- Background in robotics, reinforcement learning or other machine learning systems.
- Experience designing abstractions balancing researcher flexibility with system reliability.
Tools
JAX, PyTorch, TensorFlow, Kubernetes, SLURM, GCP, AWS
AI focus
Develop infrastructure for large-scale robot reinforcement and machine learning training.
Model-written summary · the original posting is authoritative