Senior / Staff Machine Learning Engineer - Behavior Models for Road Users
Foster City, CA
- Pay
USD 254,000–366,000/yearLocation-specific pay — pay source
Salary 254,000 – 366,000 USD per year There are three major components to compensation for this position: salary, Amazon Restricted Stock Units (RSUs), and Zoox Stock Appreciation Rights. A sign-on bonus may be offered as part of the compensation package. The listed range applies only to the base salary. Compensation will vary based on geographic location and level. Leveling, as well as positioning within a level, is determined by a range of factors, including, but not limited to, a candidate's relevant years of experience, domain knowledge, and interview performance. The salary range listed in this posting is representative of the range of levels Zoox is considering for this position.
Read the full posting- Work setting
Hybrid — work setting source
Work setup in the employer’s posting: hybrid
From the employer’s posting- Employment
Full-time — employment source
Employment type Full-time
Employment type in the employer’s posting: Full-time
From the employer’s posting
What you’ll work on
Full postingDevelop imitation and reinforcement learning models that generate human-like driving plans
Build generative behavior models conditionable on scenario intent for stress-testing
Estimate driving plan quality across safety, progress, comfort and realism
Develop metrics and tools to analyze errors and track system improvements
From the employer’s posting
Develop new deep learning models that use imitation learning and reinforcement learning to generate driving plans for human-like driving agents.
Build generative behavior models (e.g. autoregressive, diffusion) that are conditionable on scenario intent
Work on novel techniques to estimate the quality of those driving plans along the dimensions of safety, progress, comfort and realism.
Develop metrics and tools to analyze errors and understand improvements of our systems.
What you’ll bring
All qualificationsCore 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
Qualification wording
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
Education & alternatives
Qualifications - PhD degree in computer science or related field and 4+ years of relevant professional experience or master's degree and 7+ years of relevant professional experience - Experience in one of the following: Planning, Prediction, Reinforcement Learning, Imitation Learning, generative modeling (diffusion, autoregressive models)
Tools in this posting
- C++
- Python
Source — Tool mentions in context
- Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines - Fluency in Python ML frameworks and a basic understanding of C++ Bonus Qualifications
Posting record
Report an errorOpen 6 monthsWe first saw this posting 197 days ago and it is still up. Roles open this long are often slow to fill or kept open to collect applications.
- Status in our records
- Live
- First seen by us
- Apr 15, 2026 · 197 days ago
- Recorded sightings
- 317
- Last seen by us
- Oct 29, 2026
- Employer says posted
- Dec 17, 2025
These dates show when we found the posting. Check the employer’s website to confirm it is still accepting applications.
Your next step
- Have your resume and examples of relevant work ready.
- Check the listed location, eligibility and core experience before starting.
Complete your application on jobs.lever.co. The employer’s form will show what is required.
Already applied? Track this application
Job description
In this role, you will...
-
Develop new deep learning models that use imitation learning and reinforcement learning to generate driving plans for human-like driving agents.
-
Work on novel techniques to estimate the quality of those driving plans along the dimensions of safety, progress, comfort and realism.
-
Build generative behavior models (e.g. autoregressive, diffusion) that are conditionable on scenario intent — "cut off the ego vehicle," "jaywalk here" — for targeted stress-testing.
-
Leverage our compute, infrastructure and large corpus of data to push boundaries of the field.
-
Develop metrics and tools to analyze errors and understand improvements of our systems.
-
Collaborate with engineers on Perception, Planning, Simulation, and Validation to solve the overall Autonomous Driving problem.
Qualifications
-
PhD degree in computer science or related field and 4+ years of relevant professional experience or master's degree and 7+ 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++
Bonus Qualifications
-
Top tier publications (NeurIPS, ICML, CVPR)
-
Experience with JAX
Salary
254,000 – 366,000 USD per year
Employment type
Full-time
Source details
Source notes
Source excerptsSelected passages from the saved posting. Check the full description for conditions and exceptions.
- Pay
Salary 254,000 – 366,000 USD per year There are three major components to compensation for this position: salary, Amazon Restricted Stock Units (RSUs), and Zoox Stock Appreciation Rights. A sign-on bonus may be offered as part of the compensation package. The listed range applies only to the base salary. Compensation will vary based on geographic location and level. Leveling, as well as positioning within a level, is determined by a range of factors, including, but not limited to, a candidate's relevant years of experience, domain knowledge, and interview performance. The salary range listed in this posting is representative of the range of levels Zoox is considering for this position.
- Location & working pattern
Foster City, CA
Working pattern and location restrictions need checking in the full posting.
- Work authorization
No clear work-authorization passage found. Eligibility is unconfirmed.
See how this role fits your experience
Add your resume to compare the role’s scope, tools and requirements with your experience.