← All roles

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

Zoox· Software – Autonomy Software
Apply on company site
Hybrid (Foster City, CA)$254k - $366kFull-timePosted 10mo ago

About the role

Senior/Staff ML Engineer focused on developing/learning behavior models for road users (cars, bicycles, pedestrians) to populate Zoox simulations and stress-test driving systems using imitation learning, reinforcement learning, and generative modeling.

Responsibilities

  • 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++

Skills

Required:Imitation LearningReinforcement LearningGenerative Modeling (diffusion, autoregressive)Transformer-based modelsPython ML FrameworksC++

Benefits

  • paid time off (e.g. sick leave, vacation, bereavement)
  • unpaid time off
  • Zoox Stock Appreciation Rights
  • Amazon RSUs
  • health insurance
  • long-term care insurance
  • long-term and short-term disability insurance
  • life insurance
Last updated Oct 21, 2026. Interested? Apply on the company site.
Zoox logo

About Zoox

It's Not a Car | Zoox

Zoox is a purpose-built autonomous vehicle designed for riders, not drivers. Learn more about the Zoox robotaxi and the future of ride-hailing.

autonomous vehiclerobotaxiride-hailing
Company profileWebsite

More open roles at Zoox