Applied Scientist/Machine Learning Engineer Gaia

Wayve; Sunnyvale (CA); On-site; USD 311,000 - 512,000; Full time

Benefits offered by this job

Hybrid work policy

Office in Sunnyvale

Competitive compensation

Job summary

Wayve is seeking a senior Machine Learning Engineer for the Simulation team in Sunnyvale or London. You will build next‑generation generative world models and planners to simulate diverse driving environments, enabling faster training and broader testing.

Expect to optimize latency, push the boundaries of diffusion/latent video tech, and mentor researchers while collaborating across ML, sim, and deployment teams. This role blends research excellence with production impact.

Qualifications

• 4+ years of experience in ML research/engineering with a focus on generative video, world models.
• Deep knowledge in diffusion & latent‑video models; track record of improving sampling efficiency or model throughput.
• Experience working with high‑dimensional temporal or spatial‑temporal data (e.g., video, multi‑sensor fusion).
• Strong Python and PyTorch engineering fundamentals, and experience building research‑grade production tools.
• Strong publication record or contributions to open‑source ML tooling.

Responsibilities

• Invent next‑generation, efficient generative world‑models (diffusion, transformer or hybrid) delivering real‑time rollouts and controllable scene editing.
• Architect interactive world models enabling reinforcement learning, planning and safety evaluation loops.
• Optimize end‑to‑end performance, reduce inference latency by orders of magnitude.
• Define robust metrics for long‑horizon coherence, physics fidelity and planner integration; run ablations and scaling studies.
• Ship impact by integrating models into closed‑loop training and evaluation; measure sim‑to‑real gap against on‑road results.
• Mentor junior researchers, shape technical road‑maps, publish at top venues and represent Wayve in the community.
• Challenge assumptions and drive innovation: propose bold ideas, conduct ablations, and explore new training approaches.

Skills

ML research

Diffusion models

Python

PyTorch

Production tooling

Publications/Open-source

Job description

The role

Generative simulation is the team that is advancing our end-to-end autonomous driving research. The team’s mission is to accelerate our journey to AV2.0 and ensure the future success of Wayve by incubating and investing in new ideas that have the potential to become game-changing technological advances for the company.

Where You’ll Have Impact

This role would sit within Simulation focusing on unlocking disruptive innovation that solves self-driving. We believe the next leap in autonomy comes from a world model that is faithful enough, and fast enough, to train and evaluate driving models in closed loop — not only to replay logs.

As a Machine Learning Engineer in the Simulation team, you’ll play a key role in developing next-generation world models and planners that can simulate complex, diverse, and temporally consistent driving environments. These generative simulation models (like GAIA) will power faster training, broader testing, and scalable deployment—even in areas and scenarios we’ve never driven in before.

As we push toward the next generation of GAIA, efficiency and interactivity become a great focus area: models must run thousands of roll-outs per second, support closed-loop agent interaction and fit within practical compute budgets. This role will lead that leap.

You’ll work at the intersection of machine learning research, multi-modal modeling, and real‑world deployment tackling questions like:

• How can we deploy AVs in a new geography without collecting any real‑world data?
• Can synthetically generated environments fully replace physical testing and data collection?

Key Responsibilities

You will be a senior technical contributor inside Simulation, the team that incubates breakthrough ideas for Wayve.

Your mandate:

• Invent next‑generation, efficient generative world‑models (diffusion, transformer or hybrid) that deliver real‑time roll‑outs and controllable scene editing.
• Architect interactive world models where agents (or humans) can step the model, enabling reinforcement learning, planning and safety evaluation loops.
• Optimise end‑to‑end performance – from latent compression to context pruning your aim is to reduce inference latency by orders of magnitude.
• Define robust metrics for long‑horizon coherence, physics fidelity and planner integration; run ablations and scaling studies to understand trade‑offs.
• Ship impact: integrate your models into closed‑loop training and evaluation, and measure the sim‑to‑real gap against on‑road driving‑model results.
• Mentor & influence: guide junior researchers, shape technical road‑maps, publish at top venues and represent Wayve in the community
• Challenge assumptions and drive innovation: propose bold ideas, conduct ablation studies, and question conventional approaches to training and evaluation.

About You

In order to set you up for success at Wayve, we’re looking for the following skills and experience.

• 4+ years of experience in ML research/engineering with a focus on generative video, world models.
• Deep knowledge in diffusion & latent‑video models; track record of improving sampling efficiency or model throughput
• Experience working with high‑dimensional temporal or spatial‑temporal data (e.g., video, multi‑sensor fusion).
• Strong Python and PyTorch engineering fundamentals, and experience building research‑grade production tools.
• Strong publication record or contributions to open‑source ML tooling.
• Ability to work collaboratively in a fast‑paced, innovative, interdisciplinary team environment.

Desirable

• Experience in AVs, robotics, simulation, or other embodied AI domains.
• Experience working with synthetic‑to‑real transfer.

Why Join Us

• Work on transformative technology with real‑world impact on mobility, safety, and AI.
• Access massive driving datasets, cutting‑edge infrastructure, and world‑class research talent.
• Be part of a high‑trust, high‑autonomy team that values creativity, experimentation, and deep thinking.
• Publish, share, and shape the future of generative AI for autonomy.

This is a full‑time role based in our office in London or Sunnyvale. At Wayve we want the best of all worlds so we operate a hybrid working policy that combines time together in our offices and workshops to fuel innovation, culture, relationships and learning, and time spent working from home. We operate core working hours so you can determine the schedule that works best for you and your team.

For Sunnyvale, the salary ranges are between USD 311.000- 512.000 depending on experience and performance.