active · Approximate location
Machine Learning Engineer - Behavior Models for Road Users
full timeFoster City, CA
Create a free profile to personalize salary, commute and role fit.
AT A GLANCE
Technology3+ years experienceFull TimeMaster’s degree required
Required
- 3+ years experience
- Master’s degree
This role in the Offline Driving Intelligence team is responsible for developing/learning behavior models for road users such as cars, bicycles, and pedestrians. These agents populate Zoox's simulations and must be indistinguishable from real road users, yet fully controllable: p
About the role
This role in the Offline Driving Intelligence team is responsible for developing/learning behavior models for road users such as cars, bicycles, and pedestrians. These agents populate Zoox's simulations and must be indistinguishable from real road users, yet fully controllable: promptable into the rare, adversarial, safety-critical behaviors we need to test against. This means the team’s models directly impact how fast Zoox can train, validate and ship its driving stack. Our team collaborates closely with Planner, Simulation and Validation teams to develop and validate our driving performance. As an ML Agents Machine Learning Engineer, you will work on the bleeding edge of the industry, developing novel machine learning pipelines and models to predict the behavior of other agents in the world and planning the best course of action for the ego vehicle.
Requirements
- PhD degree in computer science or related field or master's degree and 3+ years of relevant professional experience
- Experience in one of the following: Planning, Prediction, Reinforcement Learning, Imitation Learning, generative modeling (diffusion, autoregressive models)
- Experience with 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++
- Top tier publications (NeurIPS, ICML, CVPR)
- Experience with JAX
Other Jobs From This Employer
Senior Vehicle Attributes EngineerFoster City, CA
Salary not listedSenior Strategic Sourcing ManagerBoston, MA
Salary not listedStaff Compensation ManagerFoster City, CA
Salary not listedStaff Software Engineer, HPCFoster City, CA
Salary not listedManufacturing Test EngineerFoster City, CA
Salary not listed