Senior / Staff Machine Learning Engineer - Behavior Models for Road Users

The role

You will develop deep learning models that predict and generate the behavior of cars, bicycles and pedestrians in Zoox's simulations. You will also work on ways to judge the quality of driving plans for safety, progress, comfort and realism.

What you will do

• Develop new deep learning models using imitation learning and reinforcement learning to generate driving plans for human-like driving agents
• Work on techniques to estimate the quality of driving plans for safety, progress, comfort and realism
• Build generative behavior models (e.g. autoregressive, diffusion) that can be conditioned on scenario intent for stress-testing
• Leverage compute, infrastructure and a large corpus of data to advance the field
• Develop metrics and tools to analyze errors and understand improvements of systems
• Collaborate with Perception, Planning, Simulation and Validation engineers on the overall autonomous driving problem

Required qualifications

• PhD in computer science or a related field and 4+ years of relevant professional experience, or a master's degree and 7+ years of relevant professional experience
• Experience in one of: Planning, Prediction, Reinforcement Learning, Imitation Learning, or generative modeling (diffusion, autoregressive models)
• Experience training and deploying transformer-based model architectures
• Experience with production machine learning pipelines: dataset creation, training frameworks, metrics pipelines
• Fluency in Python ML frameworks and a basic understanding of C++

Nice to have

• Top tier publications (NeurIPS, ICML, CVPR)
• Experience with JAX

Benefits

• Paid time off (sick leave, vacation, bereavement)
• Unpaid time off
• Zoox Stock Appreciation Rights
• Amazon Restricted Stock Units (RSUs)
• Health insurance
• Long-term care insurance
• Long-term and short-term disability insurance
• Life insurance

Before you apply

• Leveling and pay position depend on years of experience, domain knowledge and interview performance
• Compensation varies by geographic location and level

About the company

Zoox works on autonomous driving, using simulations to test and validate its driving stack. The Offline Driving Intelligence team builds behavior models that populate those simulations.