Be an Early Applicant
Zoox is an autonomous mobility company that’s created a purpose-built robotaxi to give the world a better way to ride.
The Role
Develop and deploy deep learning models (imitation and reinforcement learning) for agent behavior prediction and driving plan generation. Build large-scale ML pipelines, training frameworks, and metrics to evaluate safety, comfort, progress, and realism. Create tools to analyze errors and collaborate with Perception, Planning, Simulation, and Validation teams to improve autonomous driving performance.
Summary Generated by Built In
The Offline Driving Intelligence team is responsible for developing Foundation Models for ML Agents and planning, applying them off-vehicle to provide generalization capabilities to simulation and validation. Our team collaborates closely with the Planner, Simulation and Validation teams to develop and validate our driving performance. As an ML Agents and Planning Machine Learning Engineer you will work on the bleeding edge of the industry, developing novel machine learning pipelines and models to predict the behavior of other agents in the world and planning the best course of action for the ego vehicle.
In this role, you will...
- 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.
Qualifications
- 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++
Bonus Qualifications
- Top tier publications (NeurIPS, ICML, CVPR)
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.
Zoox also offers a comprehensive package of benefits, including 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, and life insurance.
Skills Required
- PhD in computer science or related field OR master's degree plus 5+ years of professional experience
- Experience in planning and/or prediction using reinforcement learning techniques
- Experience 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++
- Top-tier publications (NeurIPS, ICML, CVPR)
Zoox Compensation & Benefits Highlights
The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about Zoox and has not been reviewed or approved by Zoox.
-
Healthcare Strength — Healthcare is extensive, with broad medical and vision options, company‑paid disability coverage, and multiple mental‑health resources. Feedback suggests coverage breadth and auxiliary programs support a wide range of needs.
-
Parental & Family Support — Family supports include paid parental leave, additional pregnancy disability time, fertility coverage, and adoption/surrogacy assistance. Backup care and family‑oriented programs further reinforce support across life stages.
-
Wellbeing & Lifestyle Benefits — Day‑to‑day perks are robust, including free daily meals, fitness subsidies, commuter support, and on‑site amenities. Feedback suggests these lifestyle benefits enhance convenience and workplace experience, especially for office‑based roles.
Zoox Insights
Am I A Good Fit?
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.
Success! Refresh the page to see how your skills align with this role.
The Company
What We Do
Zoox is an autonomous mobility company that was founded to provide a safer, cleaner, and more enjoyable future on the road. To achieve that goal, the company has spent the past 10 years creating a purpose-built robotaxi that gives the world a better way to ride.
Why Work With Us
At Zoox, we are working to solve one of the greatest technological challenges of our generation. From the beginning, we have been focused on our goal of reimagining transportation from the ground up. We are a mission-driven community of innovators working together to create a safer, cleaner, and more enjoyable future on the road.
Gallery
