Machine Learning Engineer (Closed Loop)

Posted 5 Days Ago
Be an Early Applicant
London, Greater London, England, GBR
In-Office
Mid level
Artificial Intelligence • Transportation
The Role
Develop efficient generative world models and planners for closed-loop autonomous-driving simulation. Responsibilities include designing diffusion or transformer-based models, enabling interactive agent simulation, optimizing inference latency and compute efficiency, defining evaluation metrics, integrating models into training and evaluation workflows, measuring sim-to-real performance, mentoring researchers, and contributing to technical roadmaps and publications.
Summary Generated by Built In
About us   

Founded in 2017, Wayve is the leading developer of Embodied AI technology.  Our advanced AI software and foundation models enable vehicles to perceive, understand, and navigate any complex environment, enhancing the usability and safety of automated driving systems.

Our vision is to create autonomy that propels the world forward.  Our intelligent, mapless, and hardware-agnostic AI products are designed for automakers, accelerating the transition from assisted to automated driving. 
In our fast-paced environment big problems ignite us—we embrace uncertainty, leaning into complex challenges to unlock groundbreaking solutions. We aim high and stay humble in our pursuit of excellence, constantly learning and evolving as we pave the way for a smarter, safer future.

At Wayve, your contributions matter.  We value diversity, embrace new perspectives, and foster an inclusive work environment; we back each other to deliver impact.  

Make Wayve the experience that defines your career!  

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

Wayve is committed to creating an inclusive interview experience. If you require any accommodations or adjustments to participate fully in our interview process, please let us know.

We understand that everyone has a unique set of skills and experiences and that not everyone will meet all of the requirements listed above. If you’re passionate about self-driving cars and think you have what it takes to make a positive impact on the world, we encourage you to apply.
At Wayve we're committed to creating a diverse, fair and respectful culture that is inclusive of everyone based on their unique skills and perspectives, and regardless of sex, race, religion or belief, ethnic or national origin, disability, age, citizenship, marital, domestic or civil partnership status, sexual orientation, gender identity, veteran status, pregnancy or related condition  (including breastfeeding) or any other basis as protected by applicable law.  

For more information visit Careers at Wayve. 

To learn more about what drives us, visit Values at Wayve 

For US candidates only, please visit E-Verify Notice and Participation and Right to Work

DISCLAIMER: We will not ask about marriage or pregnancy, care responsibilities or disabilities in any of our job adverts or interviews. However, we do look to capture information about care responsibilities, and disabilities among other diversity information as part of an optional DEI Monitoring form to help us identify areas of improvement in our hiring process and ensure that the process is inclusive and non-discriminatory.



Skills Required

  • 4+ years of experience in machine learning research or engineering focused on generative video or world models
  • Deep knowledge of diffusion and latent-video models
  • Track record of improving sampling efficiency or model throughput
  • Experience with high-dimensional temporal or spatiotemporal data, such as video or multisensor fusion
  • Strong Python and PyTorch engineering fundamentals
  • Experience building research-grade production tools
  • Strong publication record or contributions to open-source machine learning tooling
  • Ability to collaborate in a fast-paced, innovative, interdisciplinary team
  • Experience in autonomous vehicles, robotics, simulation, or embodied AI
  • Experience with synthetic-to-real transfer

Wayve Compensation & Benefits Highlights

The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about Wayve and has not been reviewed or approved by Wayve.

  • Healthcare Strength Private healthcare and access to therapy via Spill are part of the standard package. This indicates robust health support within the core offering.
  • Leave & Time Off Breadth Paid vacation, public holidays, and additional leave programs are explicitly listed. This breadth of time away supports work–life balance expectations.
  • Equity Value & Accessibility Cash plus equity is standard in offers at this growth stage. This provides ownership alongside salary with perceived upside tied to company momentum.

Wayve Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: London
200 Employees
Year Founded: 2017

What We Do

We're Wayve, a leading developer of embodied intelligence for autonomous vehicles. We use AI to pioneer a next-generation approach to self-driving: AV2.0, which enables fleet operators to unlock the benefits of AV technology at scale. Founded in 2017, Wayve is made up of a diverse team of experts in machine learning and robotics. We were the first to deploy AVs on public roads with end-to-end deep learning. Today, our teams are based in London and California, and we're testing AVs in cities across the UK. Inspired by our vision for a smarter, safer, more sustainable world, we're looking for people who are passionate about building breakthrough solutions to some of the world’s most important challenges. If you're looking for an exciting opportunity with a dynamic team, get in touch!

Similar Jobs

Hybrid
London, Greater London, England, GBR
289097 Employees
Hybrid
London, Greater London, England, GBR
289097 Employees
Hybrid
London, Greater London, England, GBR
289097 Employees

Braze Logo Braze

People Business Partner (12 Month Fixed Term)

Marketing Tech • Mobile • Software
Easy Apply
Hybrid
London, Greater London, England, GBR
2000 Employees

Similar Companies Hiring

Hanover Park Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
42 Employees
Kepler  Thumbnail
Artificial Intelligence • Fintech • Software
New York, New York
9 Employees
Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account