Senior Machine Learning Engineer - Simulation Scenario Generation jobs in United States
cer-icon
Apply on Employer Site
company-logo

Zoox · 1 day ago

Senior Machine Learning Engineer - Simulation Scenario Generation

positionFoster City, CA
timeFull-time
remoteOnsite
senioritySenior Level
money$233K/yr - $290K/yr
date5+ years exp

Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market. The Simulation Scenario Generation team is looking for a ML Engineer to enable next-generation scalable AV scenario creation workflows, contributing to both immediate user impact and long-term AI-driven safety validation.

Artificial Intelligence (AI)Autonomous VehiclesMachine LearningRoboticsTransportation
check
H1B Sponsor Likelynote

Responsibilities

Contribute to tooling for AI-based scenario understanding and validation
Synthesize realistic AV simulation scenarios with dynamic (e.g., traffic) and static features
Integrate and validate LLMs/VLMs and implement other models for complex scenario generation workflows, leveraging techniques like agentic tool use
Collaborate directly with internal customers and partner teams to provide generative AI solutions for their test creation workflows
Directly contribute to the safety and reliability of Zoox's autonomous software

Qualification

Machine LearningPythonML librariesTransformer architecturesDiffusion architecturesDataset creationSystem integrationLow-latency inferenceAutonomous vehiclesRoboticsProgram synthesisFormal methodsRelevant publications

Required

MS or PhD in Computer Science, Machine Learning, or related field
5+ years of industry experience in Machine Learning
Proficiency in Python and ML libraries (PyTorch, JAX, NumPy, etc.) demonstrated through professional or research projects
Demonstrated experience in transformer and diffusion architectures
Practical experience in dataset creation for fine-tuning, system integration of ML models into production, or optimization techniques for low-latency inference systems

Preferred

Familiarity with autonomous vehicles, robotics, and/or complex simulation environments
Hands-on experience in areas like program synthesis and/or formal methods/V&V
Relevant publications in conferences (e.g., CVPR, ICCV, RSS, and/or ICRA)

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

twittertwittertwitter
Glassdoor3.8
company-logo
Zoox is an AI robotics company that provides mobility-as-a-service and self-driving car services. It is a sub-organization of Amazon.
dateFounded in 2014
location
Foster City, California, USA
people-size1001-5000 employees
web-link
https://www.zoox.com/

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 Stage
Total Funding
$955M
Key Investors
Grok Ventures
2020-06-26Acquired
2019-10-21Convertible Note· $200M
2018-07-08Series B· $465M

Leadership Team

leader-logo
Nelson Pedreiro
Sr. Vice President, Hardware
linkedin
Z
Zheng Gao
Director of Hardware Engineering
linkedin

Recent News

Channel NewsAsia
Amazon's Zoox to recall 332 US vehicles over software error
2025-12-24
AfroTech
Two Black Woman Leaders Are At The Forefront Of The Robotaxi Revolution As CEOs Of Major Vehicle Companies Waymo And Zoox
2025-12-17
Clean Technica
Zoox Becomes Robotaxi Partner for T-Mobile Arena
2025-12-13
Company data provided by crunchbase