Jobless Developer
Zoox logo
Zoox

Posted 5 months ago

Open

Senior Machine Learning Engineer - Simulation Scenario Generation

Foster City, CAOn-siteFull-time

AI Summary

Senior ML Engineer focusing on scalable scenario generation for autonomous vehicle testing, developing AI-based tools to create and validate synthetic traffic scenarios and integrating LLMs/VLMs into production workflows.

About this role

Do you enjoy applying machine learning to complex, real-world problems in autonomous vehicle testing? The Simulation Scenario Generation team is looking for a ML Engineer to enable next-generation scalable AV scenario creation workflows. This ranges from generating large-scale traffic simulations to extending our agentic AI system to assist in synthetic scenario creation from a 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:

  • 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

    Agentic Tool UseDataset CreationDiffusion ModelsFine-tuningFormal MethodsJAXLLMsLow-latency InferenceMetadata HandlingNumPyProduction ML DeploymentProgram SynthesisPythonPyTorchSimulation ToolingSynthesisTransformer ArchitecturesVLMsV&V

    Explore related jobs

    More jobs at Zoox

    Similar Agentic Tool Use jobs

    Browse these categories

    Remote JobsLLMs JobsNumPy JobsPython JobsPyTorch Jobs