Zoox · 5 months ago
Machine Learning Engineer - Prediction and Planning
Foster City, CA
Full-time
Hybrid
Mid, Senior Level
$204K/yr - $245K/yr
1+ years expZoox is a company focused on developing cutting-edge technologies for autonomous vehicles. The Prediction and Planning Machine Learning Engineer will be responsible for creating deep learning models and machine learning pipelines to enhance the vehicle's decision-making capabilities in complex environments.
Artificial Intelligence (AI)Autonomous VehiclesMachine LearningRoboticsTransportation
Responsibilities
Develop new deep learning models that use imitation learning and reinforcement learning to generate driving plans for our autonomous vehicle
You will also work on novel techniques to estimate the quality of those driving plans along the dimensions of safety, progress, comfort, compliance
Contribute to our large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field
Develop metrics and tools to analyze errors and understand improvements of our systems
Collaborate with engineers on Perception, Planning, and Simulation to solve the overall Autonomous Driving problem in complex urban environments
Qualification
Required
PhD degree in Computer Science or related field +1 year of professional experience (top tier publications can remove the need for the year of experience) OR MSc +5y 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 C++ or Fluency in Python with a basic understanding of C++
Preferred
Top tier publications (NeurIPS, ICML, CVPR)
Benefits
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
Company
Zoox
Zoox is an AI robotics company that provides mobility-as-a-service and self-driving car services. It is a sub-organization of Amazon.
1001-5000 employees
H1B Sponsorship
Zoox has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (293)
2024 (297)
2023 (209)
2022 (204)
2021 (131)
2020 (83)
Funding
Current Stage
Late StageTotal Funding
$955MKey Investors
Grok Ventures
2020-06-26Acquired
2019-10-21Convertible Note· $200M
2018-07-08Series B· $465M
Leadership Team
Nelson Pedreiro
Sr. Vice President, Hardware
Z
Zheng Gao
Director of Hardware Engineering
Recent News
Channel NewsAsia
2025-12-24
Clean Technica
2025-12-13
Company data provided by crunchbase