# Senior Machine Learning Engineer - ML Agents and Planning

**Company:** [Zoox](https://hotfix.jobs/companies/zoox)
**Location:** Foster City, CA
**Role:** ML Engineering
**Salary:** $242k – $290k/yr
**Experience:** 1+ years
**Skills:** Reinforcement Learning, Imitation Learning, Transformers, Python, C++, Machine Learning Pipelines, Deep Learning, Planning, Prediction, Autonomous Driving
**Posted:** 2025-12-17

> Develops ML models using imitation and reinforcement learning for agent behavior prediction and autonomous vehicle planning. Requires PhD +1 year or MSc +5 years experience, expertise in RL, transformers, and production ML pipelines.

## Job Description

## Responsibilities
- Develop deep learning models using imitation learning and reinforcement learning for human-like agent driving plans.
- Create novel techniques to estimate driving plan quality (safety, progress, comfort, realism).
- Contribute to large-scale ML infrastructure to discover solutions and advance the field.
- Develop metrics and tools to analyze errors and improvements.
- Collaborate with Perception, Planning, Simulation, and Validation engineers on autonomous driving.

## Qualifications
- PhD in computer science or related field +1 year professional experience (top-tier publications may substitute), or MSc +5 years professional experience.
- Experience in Planning/Prediction using Reinforcement Learning.
- Experience training and deploying transformer-based models.
- Experience with production ML pipelines: dataset creation, training frameworks, metrics.
- Fluency in Python; basic C++ understanding.

## Bonus Qualifications
- Top-tier publications (NeurIPS, ICML, CVPR).

## Similar roles

- [Senior Machine Learning Engineer - Perception Mapping](https://hotfix.jobs/jobs/49bb2893-5822-4d07-aa97-260a23af0156) - Zoox - Foster City, CA - $242k – $290k/yr
- [Senior Machine Learning Engineer - Perception](https://hotfix.jobs/jobs/82e55013-d9de-49ab-a7e2-37ae4e97c0c8) - Zoox - Foster City, CA - $242k – $290k/yr
- [Senior AI/ML Engineer](https://hotfix.jobs/jobs/feac02ca-e4ed-4189-9ee5-c17968056cf5) - Sigma - New York, NY - $240k – $270k/yr
- [Senior Machine Learning Engineer](https://hotfix.jobs/jobs/6ef84275-7cb9-42d7-b29f-cdcbe24624e0) - Attentive - San Francisco, CA - $244k – $320k/yr
- [Senior Software Engineer, AI Evals](https://hotfix.jobs/jobs/58e693b0-5569-4b9f-8a33-52bac16caeab) - Sentry - San Francisco, CA - $240k – $280k/yr

**Apply:** https://hotfix.jobs/jobs/482ee8a9-f2f2-4c5d-9ea7-f556ded7d518
**Canonical:** https://hotfix.jobs/jobs/482ee8a9-f2f2-4c5d-9ea7-f556ded7d518