Drimva

Zoox

Senior / Staff Software Engineer - Learned Trajectory Machine Learning Engineer

$242K–$349K

Foster City, CA · Hybrid · Staff · Full-time · Posted Nov 7, 2024

Education
BS, MS, or PhD in computer science or a related field

About the job

Develop deep learning models that generate and evaluate driving trajectories for autonomous vehicles, and improve their performance through analysis and collaboration.

What you'll do

  • Develop imitation- and reinforcement-learning models to generate driving plans and estimate their safety, progress, and comfort.
  • Use large-scale ML infrastructure to develop solutions; build metrics and tools to analyze errors and system improvements.
  • Collaborate with Perception, Planning, and Simulation engineers on autonomous driving in complex urban environments.

What they ask for

  • Experience training and deploying transformer-based models and reinforcement learning.
  • Experience with production ML pipelines, including dataset creation, training frameworks, and metrics pipelines.
  • Fluency in Python and a basic understanding of C++.
  • Extensive programming and algorithm-design experience, with strong mathematics skills.

Nice to have

  • Publications in machine learning or robotics venues.
  • Experience with prediction, autonomous vehicles, or robotics.

Pay

USD 242,000–349,000 per-year-salary (base salary)

Tools

Python, C++

AI focus

Develop deep learned trajectory models using imitation learning and reinforcement learning, and evaluate driving-plan quality.

Model-written summary · the original posting is authoritative

First seen Oct 10, 2026 · Last verified Nov 4, 2026 (UTC)