Zoox · 2 days ago
Senior Machine Learning Engineer - ML Agents and Planning
Foster City, CA
Full-time
Onsite
Senior Level
$242K/yr - $290K/yr
5+ years expZoox is responsible for developing Foundation Models for ML Agents and planning, applying them off-vehicle to provide generalization capabilities to simulation and validation. In this role, you will develop novel machine learning pipelines and models to predict the behavior of other agents and plan the best course of action for the ego vehicle.
Artificial Intelligence (AI)Autonomous VehiclesMachine LearningRoboticsTransportation
Responsibilities
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
Qualification
Required
PhD degree in computer science or related field +1y 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 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