Machine Learning Engineer (Autonomous Driving)

TL;DR

Machine Learning Engineer (Autonomous Driving): Developing foundation models for ML agents and planning to enhance simulation and validation capabilities with an accent on imitation and reinforcement learning. Focus on building novel deep learning pipelines, training transformer-based architectures, and optimizing driving performance for autonomous vehicles.

Location: Foster City, CA (Hybrid)

Salary: $189,000–$270,000 per year

Company

Zoox is an autonomous vehicle company developing fully autonomous, electric, purpose-built vehicles for ride-hailing.

What you will do

• Develop deep learning models using imitation and reinforcement learning to generate human-like driving plans.
• Create techniques to evaluate driving plan quality across safety, progress, comfort, and realism.
• Contribute to large-scale machine learning infrastructure and model training pipelines.
• Develop metrics and diagnostic tools to analyze system errors and performance improvements.
• Collaborate with perception, planning, simulation, and validation teams to solve autonomous driving challenges.

Requirements

• PhD in Computer Science or related field, or Master's degree with 5+ years of professional experience.
• Experience in planning or prediction using reinforcement learning techniques.
• Proficiency in training and deploying transformer-based model architectures.
• Experience with production machine learning pipelines, including dataset creation and training frameworks.
• Fluency in Python and basic understanding of C++.

Nice to have

• Top-tier research publications (NeurIPS, ICML, CVPR).

Culture & Benefits

• Comprehensive health, dental, and vision insurance.
• Long-term and short-term disability and life insurance.
• Amazon Restricted Stock Units (RSUs) and Zoox Stock Appreciation Rights.
• Generous paid time off including vacation and sick leave.
• Collaborative environment working on cutting-edge autonomous driving technology.