Senior ML Engineer - Realism & Multimodal Simulation

Neura Market; Mountain View (CA); Hybrid; USD 213,000 - 263,000; Full time

Benefits offered by this job

Bonus program

Equity incentive plan

Generous benefits

Job summary

Waymo is seeking visionary machine learning engineers and researchers to architect scalable deep learning systems, novel data workflows, and eval tools powering our research roadmap. You will define and measure realism in multimodal world models for the Waymo Driver, guiding the trajectory of autonomous simulation.

You will Mentor engineers, design robust pipelines, and collaborate with Alphabet teams to advance 4D world modeling and generative AI in production.

Qualifications

• Five or more years of experience in machine learning engineering or applied deep learning.
• Portfolio of shipped products or peer‑reviewed publications.
• Proficient in Python and modern ML frameworks.

Responsibilities

• Lead the design, development and deployment of cutting-edge evaluation approaches to assess realism of state-of-the-art multimodel world models and generative systems for simulation use cases at Waymo.
• Architect and implement robust and scalable machine learning pipelines for tuning, evaluating, and deploying large-scale discriminator models for the purposes of simulator realism evaluation.
• Evaluate open-source and production-ready video generation techniques that measure realism (e.g. temporal stability, multi-modal consistency, geometric discrepancy, condition following, etc.).
• Apply vision language models to evaluate semantic understanding and controllability across our world simulation products.
• Collaborate with research teams across Waymo and Alphabet to integrate advancements in 4D world modeling and generative AI into production systems.
• Mentor engineers on the team and provide technical guidance on architecture and execution.

Skills

Python programming

ML frameworks (Jax/Flax/PyTorch)

Evaluation frameworks

Education

Bachelor's, Master's, or PhD in computer science, machine learning, robotics, or related field

Tools

Jax

Flax

PyTorch

Job description

Waymo is seeking visionary machine learning engineers and researchers to architect scalable deep learning systems, novel data workflows, and eval tools powering our research roadmap. You will define and measure realism in multimodal world models for the Waymo Driver, guiding the trajectory of autonomous simulation.

You will Mentor engineers, design robust pipelines, and collaborate with Alphabet teams to advance 4D world modeling and generative AI in production.