Machine Learning Engineer

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

Salary: £42,000 - 42,000 per year

Requirements:

• 4+ years of experience in ML research or engineering with a focus on generative video or world models
• Deep knowledge of diffusion and latent-video models, with a track record of improving sampling efficiency or model throughput
• Experience working with high-dimensional temporal or spatiotemporal data, such as video or 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
• Desirable: experience working with synthetic-to-real transfer

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

Technologies:

• AI
• Architect
• Hardware
• Machine Learning
• PyTorch
• Python
• Robotics
• Sensor Fusion
• Support

More:

We are Wayve, founded in 2017 and a leading developer of Embodied AI technology. Our advanced AI software and foundation models help vehicles perceive, understand, and navigate complex environments, improving the usability and safety of automated driving systems. We are building intelligent, mapless, hardware-agnostic AI products for automakers to accelerate the shift from assisted to automated driving. We operate in a fast-paced, high-trust, high-autonomy environment that values diversity, creativity, experimentation, deep thinking, and continuous learning. This full-time Machine Learning Engineer role is based in our London office with a hybrid working policy that combines time in the office and workshops with working from home, and we offer core working hours so teams can choose a schedule that works best for them. We provide access to massive driving datasets, cutting-edge infrastructure, and world-class research talent.

last updated 36 week of 2026