# Senior Machine Learning Engineer - Simulation Scenario Generation

**Company:** [Zoox](https://hotfix.jobs/companies/zoox)
**Location:** Foster City, CA
**Role:** ML Engineering
**Salary:** $233k – $290k/yr
**Experience:** 5+ years
**Skills:** PyTorch, JAX, NumPy, Python, Transformers, Diffusion Models, LLMs, Vlms, Machine Learning, Simulation
**Posted:** 2025-12-19

> Develop ML models for generating realistic autonomous vehicle simulation scenarios, integrating LLMs/VLMs and agentic AI for safety validation. Requires 5+ years ML experience, MS/PhD, and proficiency in PyTorch, transformers, diffusion models.

## Job Description

## 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.

## Qualifications
- 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

## Bonus Qualifications
- 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**)

## Similar roles

- [Technical Lead Manager - Perception, Self-Driving Systems](https://hotfix.jobs/jobs/435a1f5b-0b25-4e98-9155-ad4bf36b2727) - Applied Intuition - Sunnyvale, CA - $232k – $298k/yr
- [Senior AI Engineer, Agentic Data Enrichment](https://hotfix.jobs/jobs/b8c6a9c2-b17f-4be4-af78-3f9ac0b761d6) - Baselayer - San Francisco, CA - $230k – $340k/yr
- [Technical Lead Manager](https://hotfix.jobs/jobs/459dd304-2f00-4641-bc79-7f32e42e51c1) - Instabase - San Francisco, CA - $230k – $270k/yr
- [Senior Machine Learning Engineer](https://hotfix.jobs/jobs/29ce42de-a923-433c-a975-ea7af29ca814) - Otter - Mountain View, CA - $230k – $265k/yr
- [Senior Research Engineer - Safety Tooling and Data](https://hotfix.jobs/jobs/ca7e0d7f-a90f-4333-8437-612d086669af) - Cohere - Remote - $230k – $380k/yr

**Apply:** https://hotfix.jobs/jobs/cc353e3f-78f9-474a-a19f-10e382c5ec1a
**Canonical:** https://hotfix.jobs/jobs/cc353e3f-78f9-474a-a19f-10e382c5ec1a