Principal Software Engineer, ML Infrastructure

Zoox, a pioneering company in autonomous transportation, is seeking a Principal Software Engineer to lead their ML Infrastructure team. This role is crucial in their mission to revolutionize transportation through autonomous robotaxis. The position offers an opportunity to shape the future of autonomous driving technology while working with cutting-edge ML and CV systems.

The role involves leading the development of next-generation ML Infrastructure that will significantly reduce development and deployment time for large-scale ML and Foundational models. You'll be responsible for managing data infrastructure handling petabytes of sensor data, operating compute infrastructure across thousands of GPUs, and overseeing ML tools and frameworks for both in-vehicle and off-vehicle applications.

Working at Zoox means joining a company at an exciting stage of deploying robotaxis, where your impact can be substantial. You'll collaborate with various AI teams, including Perception, Prediction, Planner, Simulation, and Collision Avoidance, pushing the boundaries of ML implementation in autonomous vehicles.

The compensation package is highly competitive, ranging from $373,000 to $448,000 in base salary, plus Amazon RSUs and Zoox Stock Appreciation Rights. The company offers comprehensive benefits including various insurance options and paid time off. The hybrid work environment in Foster City, CA, provides flexibility while maintaining collaborative opportunities.

This role is perfect for a deeply technical, influential leader who can both architect complex systems and mentor teams. You'll need expertise in ML infrastructure, distributed systems, and modern ML frameworks, making this an ideal position for someone wanting to make a significant impact in autonomous vehicle technology while leading and growing a team of talented engineers.