GenAI World Models Engineer for Real-Time AV Simulation

OpenDigital Limited; Greater London; Hybrid; GBP 235,000 - 387,000; Full time

Job summary

Wayve is seeking a Machine Learning Engineer to advance our Simulation team, developing scalable world models and planners for autonomous driving. You will drive end-to-end research-to-deployment, producing efficient, real-time roll-outs and contributing to open research for the community.

Based in London or Sunnyvale, this full-time role leverages a hybrid policy to balance office collaboration with focused work from home.

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.
• Ability to work collaboratively in a fast-paced, innovative, interdisciplinary team environment.

Responsibilities

• 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.

Skills

Generative modeling

World models

Diffusion

High-dimensional data

Team collaboration

Tools

Python

PyTorch

Job description

Wayve is seeking a Machine Learning Engineer to advance our Simulation team, developing scalable world models and planners for autonomous driving. You will drive end-to-end research-to-deployment, producing efficient, real-time roll-outs and contributing to open research for the community.

Based in London or Sunnyvale, this full-time role leverages a hybrid policy to balance office collaboration with focused work from home.