Senior / Staff Machine Learning Engineer - Behavior Models for Road Users
Core
Develop behavior models for road users (cars, bicycles, pedestrians) to populate Zoox's simulations, enabling controllable, indistinguishable agents for safety-critical testing and training the driving stack.
Role type
Senior IC machine learning engineer (generative behavior models)
Builds
Generative behavior models (autoregressive, diffusion) and deep learning pipelines for autonomous driving simulation
Domain
Autonomous driving, simulation, machine learning
Deliverable
production ML models
Required skills
Imitation learning, reinforcement learning, generative modeling (diffusion, autoregressive), transformer-based architectures, production ML pipelines, Python ML frameworks
Preferred skills
JAX, top-tier publications (NeurIPS, ICML, CVPR)
Technologies
Python, C++, JAX, transformers
Responsibilities
Develop deep learning models using imitation and reinforcement learning; estimate driving plan quality (safety, progress, comfort, realism); build conditionable generative behavior models; develop metrics and tools for error analysis; collaborate with Perception, Planning, Simulation, and Validation teams
Seniority
Senior, hands-on IC