Machine Learning Engineer – ML Agents, Planning

Zoox

Machine Learning Engineer developing Foundation Models for ML Agents and collaborating on Autonomous Driving technologies. Working on innovative ML pipelines and models to enhance driving simulations.

Posted 7/23/2026full-timeFoster City • California • 🇺🇸 United StatesMid-LevelSenior💰 $189,000 - $270,000 per yearWebsite

Core Competencies

Role fit

Core Competencies

Use this summary to align your resume positioning with the role.

Expertise in developing deep learning models for autonomous driving, with a focus on imitation learning and reinforcement learning. Proficient in creating and analyzing machine learning pipelines, metrics, and tools to enhance system performance and safety.

Highest-signal resume keywords

PhD In Computer ScienceReinforcement Learning TechniquesTransformer-Based Model ArchitecturesProduction Machine Learning PipelinesFluency In Python

ATS Keywords

Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills

Deep Learning ModelsImitation LearningReinforcement LearningPlanningPredictionMetrics DevelopmentError AnalysisDataset CreationTraining FrameworksModel Deployment

Tools & Technologies

Machine Learning InfrastructurePythonC++

Industry Keywords

Autonomous DrivingSafetyComfortRealismCollaboration

Tech Stack

Tools & technologies

Python

About the role

Key responsibilities & impact

You will develop new deep learning models that use imitation learning and reinforcement learning to generate driving plans for human-like agents.
You will work on novel techniques to estimate the quality of those driving plans along the dimensions of safety, progress, comfort and realism.
You will contribute to our large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field
You will develop metrics and tools to analyze errors and understand improvements of our systems
You will collaborate with engineers on Perception, Planning, Simulation, and Validation to solve the overall Autonomous Driving problem.

Requirements

What you’ll need

PhD degree in computer science or related field or master's degree and 5+ years of professional experience in a relevant field.
Experience in Planning and / or Prediction using Reinforcement Learning techniques
Experience with training and deploying transformer-based model architectures
Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines
Fluency in Python with a basic understanding of C++

Benefits

Comp & perks

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