Software Engineer, ML Platform (ML Training)

• Build the Zoox Training framework leveraged by all ML teams, ensuring high scalability, reliability, and efficiency.
• Design, implement, and operate a robust ML platform for training, validation, serving, and monitoring of ML models.
• Collaborate with cross-functional teams including ML researchers, software engineers, and data engineers to define requirements and align on architectural decisions.

This role directly enables the development of autonomous driving technology, making transportation safer, cleaner, and more accessible for everyone.

• 2+ years of ML infrastructure experience.
• Experience with training frameworks like PyTorch, DeepSpeed, JAX, Ray, etc.
• Experience working with cloud providers like AWS.

Compensation and perks not specified in the posting. Zoox offers a dynamic work environment at the intersection of robotics, machine learning, and design.

Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem to bring this technology to market, aiming to provide next-generation mobility-as-a-service in urban environments.