Zoox · 1 month ago
Senior Machine Learning Engineer, Autonomy Validation
Foster City, CA
Full-time
Hybrid
Senior Level
$229K/yr - $276K/yr
5+ years expZoox is on an ambitious journey to develop a full-stack autonomous mobility solution for cities and safely deploy a robotaxi service. They are seeking a Senior Machine Learning Engineer to improve automated testing and validation processes by applying cutting-edge machine learning techniques. This role will involve working with large datasets and cross-functional teams to enhance the efficiency and safety of their validation efforts.
Artificial Intelligence (AI)Autonomous VehiclesMachine LearningRoboticsTransportation
Responsibilities
Lead Technical Initiatives: You will apply modern machine learning, including advanced techniques like encoder-decoder models, to critical validation problems at the intersection of ML and data science. You will serve as a key contributor and tech lead on a small, focused team
Improve Model Interpretability: You'll pioneer methods to understand the internal workings of our machine learning models, bridging the gap between "black box" models and systems engineering. This work will be crucial for building trust and ensuring the safety of our autonomous systems
Improve Feature Representation: You'll extend and refine the features and embedding space used by our models to better identify and cluster interesting driving scenarios. This involves leveraging a deep understanding of the autonomous driving stack and applying first-principles thinking to derive new, impactful features
Integrate AV Performance Data: You'll incorporate metrics and information on autonomous vehicle (AV) performance into the model to make its risk predictions more accurate and relevant
Optimize with Data Science: You'll apply your data science expertise to optimize models and sampling methodologies. This includes analyzing large datasets to identify patterns and critical insights that improve the model's ability to find risky or valuable scenarios for simulation
Collaborate Cross-Functionally: You'll work closely with system safety, data science, software, and fleet operations teams to understand their needs and integrate improvements that directly support our validation efforts
Qualification
Required
A PhD in a relevant field and/or 5+ years of experience working with machine learning models and data science methodologies in an industry setting
Expertise in machine learning concepts, including model training, evaluation, and optimization
Strong programming skills in Python and experience with relevant machine learning libraries (e.g., PyTorch, TensorFlow, Jax)
Experience with large-scale data processing and distributed computing
Experience in robotics, autonomous vehicles, or a related field, with an understanding of challenges in perception, prediction, and planning
Proven ability to drive progress independently, lead technical projects, and apply critical thinking to solve practical problems
Excellent communication skills and the ability to work effectively with cross-functional teams
Preferred
Real-world impact as demonstrated in publications, patents, presentations, blog posts, etc
Familiarity with encoder-decoder or foundation models for prediction and planning
Experience with test scripting and data analysis languages like SQL
Experience with techniques for machine learning model interpretability and explainability
Familiarity with the challenges of fleet data collection and validation in the autonomous vehicle space
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