Skip to content
ZooxZooxFoster City, CA

Senior Machine Learning Engineer - Simulation Scenario Generation

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.

233k – 290k/yr
On-site5+ YOEML Engineering

About the role

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)

Skills

PyTorchJAXNumPyPythonTransformersDiffusion ModelsLLMsVlmsMachine LearningSimulation

Similar roles

ML Engineering jobs
Applied Intuition

Technical Lead Manager - Perception, Self-Driving Systems

Applied IntuitionSunnyvale, CA

Leads team developing and deploying a unified camera-first perception model for self-driving systems across diverse vehicles, geographies, and conditions. Hands-on with ML architecture, training, evaluation, embedded optimization, and customer requirements. Requires 5+ years ML perception experience and 2+ years team leadership.

232k – 298k/yr
On-site5+ YOEML Engineering
Baselayer

Senior AI Engineer, Agentic Data Enrichment

BaselayerSan Francisco, CA

Senior AI Engineer owning end-to-end LLM-driven agent systems for business data enrichment, web presence verification, entity linking, and risk scoring at Baselayer. Requires production LLM agents experience, async Python, browser automation, and eval frameworks.

230k – 340k/yr
Hybrid5+ YOEML Engineering
Instabase

Technical Lead Manager

InstabaseSan Francisco, CA

Player-coach Technical Lead Manager for AI Systems & Agents at Instabase. Hands-on architect and builder of stateful multi-turn agent loops, secure code execution sandboxes, tool orchestration via MCP, and evaluation harnesses while leading and growing a small elite team of AI engineers.

230k – 270k/yr
Hybrid8+ YOEML Engineering
Otter

Senior Machine Learning Engineer

OtterMountain View, CA

Lead projects building and deploying large-scale ASR, NLP, and LLM systems for meeting intelligence. Requires 5+ years building production ML systems with PyTorch/JAX and experience with speech/language models.

230k – 265k/yr
Hybrid5+ YOEML Engineering
Cohere

Senior Research Engineer - Safety Tooling and Data

CohereNew York, NY

Senior Research Engineer building data synthesis, analysis, and management tooling for AI safety model training and evaluation. Requires strong software engineering, statistics, and ML framework expertise.

230k – 380k/yr
Remote5+ YOEML Engineering