Senior or Staff Machine Learning Engineer (Perception Attributes)

Join Zoox, a revolutionary company in the field of autonomous mobility. As a Senior or Staff Machine Learning Engineer, you will be a key member of the Attributes team within the Perception department. Your work will focus on developing and enhancing sophisticated behavioral models for various road users, including vehicles, pedestrians, and cyclists. You will collaborate closely with the Prediction and Planning teams to optimize your models' outputs, directly influencing how our autonomous vehicles make real-time driving decisions. This is an exciting opportunity to shape the future of urban mobility and contribute to the safety and effectiveness of our autonomous driving system.

• Lead the development of sophisticated behavioral models for vehicles, pedestrians, and cyclists as a key member of the Attributes team within Zoox's Perception department.
• Create and maintain perception attribute models that generate essential signals, enabling our autonomous vehicles to understand and predict the behavior of various road users.
• Collaborate closely with Prediction and Planning teams to optimize your models' outputs, directly influencing how our autonomous vehicles make real-time driving decisions.
• You do not need to match every listed expectation to apply for this position
• MS/PhD in computer science or related fields with a minimum of 7 years of relevant experience
• Experience with knowledge distillations from large foundation models
• Fluency programming in Python and extensive experience with algorithm design
• Strong mathematics skills
• Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines
• Experience with training and deploying Deep Learning models
• Familiarity of VLMs/VLAs/ViTs
• Experience with large model distillation in a production environment
• Familiarity with C++