Machine Learning / Reinforcement Learning Infrastructure Engineer

Roboticsconnected; Boston (MA); Presencial; USD 140.000 - 210.000; Tempo integral

Resumo da vaga

Eka Robotics is seeking a Reinforcement/Machine Learning Infrastructure Engineer to design, implement, and maintain large-scale model training systems that power next-generation robot learning. You will focus on developer experience, creating intuitive tooling and scalable infrastructure across experiments.

You will collaborate with researchers to ensure infrastructure scales from prototyping to production training runs, impacting research cycles and deployment speed.

Qualificações

• BS, MS or higher in Computer Science, Computer Engineering, Machine Learning or a related technical field.
• Strong software engineering fundamentals with a track record of building ML training infrastructure or scalable systems.
• Hands-on experience with large-scale training using JAX (preferred), PyTorch, or TensorFlow.
• Familiarity with distributed training, multi-host setups, data pipelines, and cloud/orchestration systems (Kubernetes, SLURM, GCP, AWS).
• Strong cross-functional communication, ownership mindset, and tooling for developers.
• Experience building automated testing pipelines, CI/CD for ML workflows, and custom logging/telemetry stacks.

Responsabilidades

• Own Training Infrastructure: design, implement, and maintain robust systems for large-scale model training, including orchestration and experiment tracking.
• Developer Experience & Tooling: build streamlined abstractions for launching, monitoring, debugging, and reproducing experiments.
• Scale Distributed Training: work with researchers to scale RL and ML pipelines across compute clusters.
• Resource Management: ensure efficient allocation of cloud compute resources for future scaling.
• Collaborate with Researchers: translate researchers' needs into scalable infrastructure and contribute to core model/training code.

Conhecimentos

ML training infra

Distributed systems

DevOps

Cross-functional collaboration

Software engineering fundamentals

Formação

BS in CS/CE/ML or related

MS or higher in a technical field

Ferramentas

JAX

PyTorch

TensorFlow

Kubernetes

GCP/AWS

Descrição da vaga

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 now hiring to scale our R&D effort. We are looking for hands-on individuals who are excited to help shape the future of robotics.

The Role

We are looking for a 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.

We believe that world-class infrastructure is the foundation for moving research into production. 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 directly 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.

Responsibilities

• 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.

Minimum Qualifications

• 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 TensorFlow.
• 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 & DevOps: Experience building automated testing pipelines, CI/CD for ML workflows, and custom logging/telemetry stacks.

Preferred Qualifications

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