The role involves integrating and optimizing machine learning models for autonomous vehicle scenario generation, collaborating on AI-driven safety validation, and contributing to data-driven metrics and tooling.
Do you enjoy applying machine learning to complex, real-world problems in autonomous vehicle testing? The Simulation Scenario Authoring team owns the formats and tools used to create synthetic simulation scenarios. We are looking for a hands-on ML Engineer to integrate, implement, and optimize our next-generation AV scenario generation workflow. This ranges from extending our AI assistant to advancing toward full scenario creation automation from natural language test specification. This role offers a unique chance to deliver immediate user impact while contributing to long-term AI-driven safety validation.
In this role, you will:
- Integrate and validate LLMs/VLMs and implement other models for complex scenario generation workflows, leveraging techniques like advanced prompting, agentic tool use, and more.
- Contribute to tooling for AI-based scenario understanding and validation.
- Collect data and design metrics to drive business intelligence, product iteration, and model fine-tuning.
- 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.
Qualifications
- MS or PhD in Computer Science, Machine Learning, or related field
- 2+ years of industry experience in Machine Learning
- Solid understanding of LLM or NLP concepts
- Proficiency in Python and ML libraries (PyTorch, NumPy) demonstrated through professional or research projects
Bonus Qualifications
- Practical experience in dataset creation for fine-tuning, system integration of ML models into production, or optimization techniques for low-latency inference systems
- Familiarity with autonomous vehicles, robotics, and/or complex simulation environments
- Hands-on experience in areas like program synthesis, diffusion models, and/or formal methods/V&V
- Relevant publications in conferences (e.g., CVPR, ICCV, RSS, and/or ICRA)
About Zoox
Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market. Sitting at the intersection of robotics, machine learning, and design, Zoox aims to provide the next generation of mobility-as-a-service in urban environments. We’re looking for top talent that shares our passion and wants to be part of a fast-moving and highly execution-oriented team.
Follow us on LinkedIn
Accommodations
If you need an accommodation to participate in the application or interview process please reach out to [email protected] or your assigned recruiter.
A Final Note:
You do not need to match every listed expectation to apply for this position. Here at Zoox, we know that diverse perspectives foster the innovation we need to be successful, and we are committed to building a team that encompasses a variety of backgrounds, experiences, and skills.
Top Skills
Numpy
Python
PyTorch
Zoox Foster City, California, USA Office
4000 E 3rd Ave, Foster City, CA, United States, 94404
Zoox Foster City, California, USA Office
1149 Chess Drive, Foster City, CA, United States, 94404
Zoox Fremont, California, USA Office
47540 Kato Road, Fremont, CA, United States, 94538
Zoox San Francisco, California, USA Office
60 Broadway St, San Francisco, CA, United States, 94111
Similar Jobs
Artificial Intelligence • Machine Learning • Robotics • Software • Transportation • Design • Manufacturing
As a Senior Machine Learning Engineer, you will create scalable simulation scenarios for autonomous vehicle testing, integrating AI solutions and collaborating with stakeholders.
Top Skills:
JaxNumpyPythonPyTorch
Aerospace • Hardware • Robotics • Software • Manufacturing
The Sr Dynamics Engineer owns dynamic loads for hardware on Terran R, collaborates with partners, develops load predictions, and ensures compliance with dynamics specifications.
Top Skills:
FemapMatlabNastranPython
Fintech • Financial Services
The Senior Product Marketing Manager will lead marketing strategies for institutional audiences, focusing on engagement with private market participants and aligning product capabilities with market needs. They will create narratives, partner with cross-functional teams, and drive market validation for liquidity solutions.
Top Skills:
Data AnalysisLiquidity SolutionsMarket DynamicsProduct MarketingRegulatory Requirements
What you need to know about the San Francisco Tech Scene
San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.
Key Facts About San Francisco Tech
- Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Google, Apple, Salesforce, Meta
- Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
- Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
- Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine
