Generative ML Engineer — Real-Time World Models

wayve; Greater London; Hybrid; GBP 38,000 - 46,000; Full time

Job summary

Wayve, based in London, is seeking a Machine Learning Engineer to advance generative world models and real-time rollouts in our Embodied AI research. The role demands deep diffusion knowledge, strong PyTorch/Python fundamentals, and a track record of research-grade tooling.

The position is hybrid with our London office and flexible work arrangements. You will collaborate across a fast-paced interdisciplinary team, aiming to reduce inference latency and enable real-world perception and planning

Qualifications

• 4+ years of experience in ML research or engineering focused on generative video or world models.
• Deep knowledge of diffusion and latent-video models, with improved sampling throughput.
• Strong Python and PyTorch fundamentals; experience building research-grade tools.
• Strong publication record or contributions to open-source ML tooling.
• Ability to work collaboratively in a fast-paced, interdisciplinary team environment.

Responsibilities

• Invent next-generation, efficient generative world models using diffusion, transformer, or hybrid approaches 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, to reduce inference latency by orders of magnitude.
• Define robust metrics for long-horizon coherence, physics fidelity, and planner integration, and run ablations and scaling studies to understand trade-offs.
• Ship impact by integrating models into closed-loop training and evaluation, and measure the sim-to-real gap against on-road driving-model results.
• Mentor and influence junior researchers, shape technical road maps, publish at top venues, and represent Wayve in the community.
• Challenge assumptions and drive innovation by proposing bold ideas, conducting ablation studies, and questioning conventional approaches to training and evaluation.

Skills

ML research

Generative video

World models

Team collaboration

Tools

PyTorch

Python

Job description

Wayve, based in London, is seeking a Machine Learning Engineer to advance generative world models and real-time rollouts in our Embodied AI research. The role demands deep diffusion knowledge, strong PyTorch/Python fundamentals, and a track record of research-grade tooling.

The position is hybrid with our London office and flexible work arrangements. You will collaborate across a fast-paced interdisciplinary team, aiming to reduce inference latency and enable real-world perception and planning