Machine Learning​/Reinforcement Learning Infrastructure Engineer

Eka Robotics

Eka Robotics is on a mission to build intelligence for the physical world - robots that are fast, general, and reliable. Our approach, grounded in physics, unlocks superhuman capabilities. We are defining the frontier of robotics research and deployment.

Our team consists of pioneers in robotics and machine learning. We are hiring to scale our R&D effort. We are looking for hands‑on individuals who are excited to help shape the future of robotics.

Reinforcement/Machine Learning Infrastructure Engineer to shape our training infrastructure.

In this role, you will be responsible for designing, implementing, and maintaining the large‑scale model training systems that power our next generation of robot learning. You will focus on building an exceptional developer experience, creating intuitive and efficient tooling that our engineers and scientists love to use. Your work will accelerate our research cycles, making it effortless to test new ideas and scale successful experiments into production training runs.

You will work closely with researchers to ensure our infrastructure scales seamlessly from prototyping to large‑scale distributed training. This is a hands‑on, high‑impact role at the intersection of machine learning, software engineering, and scalable infrastructure.

• Own Training Infrastructure:
  Design, implement, and maintain robust systems for large‑scale model training, including job orchestration, scheduling, checkpointing, and experiment tracking.
• Developer Experience & Tooling:
  Build streamlined, intuitive abstractions for launching, monitoring, debugging, and reproducing experiments, minimizing friction and maximizing productivity for our research teams.
• Scale Distributed Training:
  Work closely with researchers to reliably scale reinforcement learning and machine learning pipelines across compute clusters.
• Resource Management:
  Ensure efficient allocation and utilization of cloud‑based compute resources while building the foundational systems needed for future scaling.
• Collaborate with Researchers:
  Partner with the research team to understand their needs, build infrastructure that supports cutting‑edge methods, guide best practices for training at scale, and contribute to core JAX model and training code.

• Education:
  
  BS, MS or higher in Computer Science, Computer Engineering, Machine Learning or a related technical field.
• Software Engineering:
  Strong software engineering fundamentals with a proven track record of building ML training infrastructure, internal developer platforms, or scalable systems.
• Deep Learning Frameworks:
  Hands‑on experience with large‑scale training using JAX (preferred), PyTorch, or Tensor Flow.
• Distributed Systems:
  Familiarity with distributed training, multi‑host setups, data pipelines, and managing workloads on cloud platforms or orchestration systems (e.g., Kubernetes, SLURM, GCP, AWS).
• Communication & Ownership:
  Strong cross‑functional communication skills, a deep ownership mindset, and a passion for building tools that improve the developer experience.
• Infrastructure & Dev Ops:
  Experience building automated testing pipelines, CI/CD for ML workflows, and custom logging/telemetry stacks.

• Domain
  
  Experience:
  
  Background in robotics, reinforcement learning or other machine learning systems.
• Systems Design:
  Experience designing abstractions that balance researcher flexibility with system reliability.