Senior Machine Learning Engineer - Simulation Scenario Generation

About the role

The Simulation Scenario Generation team is seeking a ML Engineer to develop scalable AV scenario creation workflows. This role involves enabling AI-based scenario understanding and validation, synthesizing realistic AV simulation scenarios, integrating and validating LLMs/VLMs, and implementing other models for complex scenario generation workflows.

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
• Demonstrated proficiency in Python and ML libraries (PyTorch, JAX, NumPy, etc.) through professional or research projects
• Experience with 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)

Pay

$233,000 - $290,000 a year Base Salary Range

Schedule

Not specified

Benefits

Comprehensive benefits package including paid time off, Zoox Stock Appreciation Rights, Amazon RSUs, health insurance, long-term care insurance, long-term and short-term disability insurance, and life insurance.