Software Engineer, Machine Learning

Zoox

Software Engineer developing deep learned models for autonomous vehicles at Zoox. Collaborating with teams on behavior prediction and motion planning in urban environments.

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

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 our autonomous vehicle.
You will work on techniques to estimate the quality of those driving plans along the dimensions of safety, progress, comfort etc.
You will leverage 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, and Simulation to solve the overall Autonomous Driving problem in complex urban environments.

Requirements

What you’ll need

BS, MS, or PhD degree in computer science or related field
Experience with training and deploying transformer-based model architectures and reinforcement learning
Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines
Fluency in C++ or Fluency in Python with a basic understanding of C++
Extensive experience with programming and algorithm design
Strong mathematics skills
Conference or Journal publications in Machine Learning or Robotics related venues (bonus)
Prior experience with Prediction and/or autonomous vehicles or robotics in general (bonus)

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

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 & Tools

deep learningimitation learningreinforcement learningtransformer-based model architecturesproduction Machine Learning pipelinesdataset creationtraining frameworksmetrics pipelinesC++Python

Soft Skills

collaborationproblem-solvinganalytical skillscommunication