Zoox
Senior / Staff Machine Learning Engineer - Behavior Models for Road Users
$254K–$366K
Foster City, CA · Hybrid · Staff · Full-time · Posted Dec 17, 2025
- Experience
- PhD and 4+ years of relevant professional experience, or master's degree and 7+ years of relevant professional experience.
- Education
- PhD degree in computer science or related field, or master's degree.
About the job
Develop controllable, human-like behavior models for simulated road users to support autonomous-driving training, validation, and stress testing.
What you'll do
- Develop deep learning models using imitation and reinforcement learning to generate driving plans for simulated agents.
- Build generative behavior models conditioned on scenario intent and develop methods to evaluate plan quality and system errors.
- Collaborate across Perception, Planning, Simulation, and Validation to improve autonomous-driving performance.
What they ask for
- Experience in Planning, Prediction, Reinforcement Learning, Imitation Learning, or generative modeling.
- Experience training and deploying transformer-based model architectures.
- Experience with production machine learning pipelines, including dataset creation, training frameworks, and 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.
Pay
USD 254,000–366,000 per year, base salary
Tools
Python, C++, transformer-based model architectures, JAX, diffusion, autoregressive models
AI focus
Develop deep learning and generative models using imitation learning and reinforcement learning to simulate controllable road-user behavior, generate driving plans, and evaluate safety, progress, comfort, and realism.
Model-written summary · the original posting is authoritative