Software Engineer, ML Platform (ML Training)
Core
Build and operate the core ML platform enabling model training at scale, including deep learning frameworks and distributed training infrastructure for autonomous driving.
Role type
Senior IC ML Platform Engineer (Training Infrastructure)
Builds
Scalable ML training frameworks, distributed model training infrastructure, and model lifecycle management tools for autonomous driving.
Domain
Autonomous driving / ML Infrastructure
Deliverable
infrastructure
Required skills
Distributed model training, Deep learning frameworks, Cloud infrastructure, ML platform design, System scalability, Model lifecycle management
Preferred skills
Large-scale cost-efficient distributed training, Reinforcement learning infrastructure, Experimentation tools
Technologies
PyTorch, DeepSpeed, JAX, Ray, AWS
Responsibilities
Build the Zoox Training framework leveraged by all ML teams; Design, implement, and operate a robust ML platform for training, validation, serving, and monitoring; Collaborate with cross-functional teams to define requirements and align on architectural decisions.
Seniority
Mid-level, hands-on IC
