Senior Machine Learning Engineer, Scaling World Models

Reposted 6 Days Ago
3 Locations
In-Office
Senior level
Artificial Intelligence • Transportation
The Role
Seeking a Senior Machine Learning Engineer to develop advanced AI systems, managing datasets, enhancing performance, and collaborating across teams to innovate in self-driving technology.
Summary Generated by Built In

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.  

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 

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

The goal of this role is to build, scale, and optimise next-generation world model architectures (e.g. GAIA and successors) and bridge them into high-throughput training infrastructure, enabling synthetic data and simulation to dramatically accelerate autonomy development.

You will live at the intersection of model research, large-scale ML systems, and real-world deployment. You will both invent new generative architectures and make them trainable at scale (efficiently and reliably) so that synthetic environments can exceed reality in utility.

Key responsibilities:

  • Design and implement performance improvements (tensor parallelism, pipeline parallelism etc) for large scale training.
  • Profile and diagnose large-scale model training jobs to identify bottlenecks (GPU/compute, memory, I/O, communication) and optimise performance.
  • Train large-scale temporal models on multi-modal data (video, LiDAR, vehicle telemetry), learning representations of complex real-world dynamics.
  • Design experiments to understand model generalization, scaling behavior, and performance trade-offs between synthetic and real data.
  • Define and track metrics and benchmarks for long-horizon prediction, scene fidelity, and planner integration.
  • Challenge assumptions and drive innovation: propose bold ideas, conduct ablation studies, and question conventional approaches to training and evaluation.
  • Collaborate with platform/engineering teams to align research prototypes with production-level infrastructure.

About you:  

In order to set you up for success as an Applied Scientist at Wayve, we’re looking for the following skills and experience.  

  • Established background in ML engineering or applied research.
  • Hands-on experience optimizing large-scale training workloads (multi-GPU / multi-node), including parallelism, kernel-level optimizations, memory and I/O bottlenecks.
  • Proven experience working cross-functionally between research teams and platform / infrastructure teams.
  • Demonstrated background 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.
  • Ability to take bold ideas, run experiments, and iterate quickly.
  • Ability to work collaboratively in a fast-paced, innovative, interdisciplinary team environment.

Desirable:

  • Deep knowledge of generative modelling (e.g., auto-regressive, diffusion, or VAEs)
  • Experience in AVs, robotics, simulation, or other embodied AI domains.

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.

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.

For more information visit Careers at Wayve. 

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

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.



Top Skills

Cuda
Mlops
Python
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

Bose Logo Bose

Technical Support

Automotive • eCommerce • Hardware • Music • Retail • Software • Wearables
Hybrid
Chatham, Kent, England, GBR
2900 Employees

Bose Logo Bose

Technical Support

Automotive • eCommerce • Hardware • Music • Retail • Software • Wearables
Hybrid
Chatham, Kent, England, GBR
2900 Employees

Bose Logo Bose

Operations Manager

Automotive • eCommerce • Hardware • Music • Retail • Software • Wearables
Hybrid
London, Greater London, England, GBR
2900 Employees

BlackRock Logo BlackRock

Java Full-Stack Software Engineering Team Leader, Vice President

Fintech • Information Technology • Financial Services
In-Office
London, Greater London, England, GBR
25000 Employees

Similar Companies Hiring

Credal.ai Thumbnail
Software • Security • Productivity • Machine Learning • Artificial Intelligence
Brooklyn, NY
Standard Template Labs Thumbnail
Software • Information Technology • Artificial Intelligence
New York, NY
10 Employees
Scotch Thumbnail
Software • Retail • Payments • Fintech • eCommerce • Artificial Intelligence • Analytics
US
25 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account