Machine Learning Engineer

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, and we encourage people to apply even if they do not meet every requirement listed.

last updated 36 week of 2026