Hybrid Senior Machine Learning Engineer – Simulation Scenario Generation

Posted 3 weeks ago

Apply now

About the role

  • ML Engineer in Simulation Scenario Generation team applying machine learning for autonomous vehicle testing. Delivering immediate user impact while contributing to long-term AI-driven safety validation.

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.

Requirements

  • 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

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

Job title

Senior Machine Learning Engineer – Simulation Scenario Generation

Job type

Experience level

Senior

Salary

$233,000 - $290,000 per year

Degree requirement

Postgraduate Degree

Location requirements

Report this job

See something inaccurate? Let us know and we'll update the listing.

Report job