Latest AI JobsBrowse Jobs
wayve logo

Principal Machine Learning Engineer GAIA

wayveLondon, United KingdomSimulation, Evaluation, Validation
Apply on company site
Salary
Not listed
Job type
Full-time
Work mode
Remote
Experience
Senior · 5+ yrs
Posted
6 Oct 2026
Status
Open · checked 7d ago

About this role

This role suits experienced machine learning engineers with a background in large-scale model fine-tuning and world models who want to shape embodied intelligence for autonomous vehicles in London. Wayve is hiring a Principal Machine Learning Engineer for GAIA to work on the simulation evaluation and validation team in London, United Kingdom.

Role Overview

The Principal Machine Learning Engineer GAIA position is a full-time role based in London, United Kingdom, operating on a hybrid working model. The position involves working on Gaia, which is Wayve's world model trained on large-scale driving video. The team focuses on simulation evaluation and validation, partnering with research, applications, core simulation engineering, and cloud infrastructure teams to translate model improvements into downstream impacts on the driving stack.

Key Responsibilities

Own and lead parts of large-scale language- or video-style training for Gaia from experimentation through productionised training runs.
Contribute to and influence model architecture decisions beyond applying existing models.
Partner across research, applications, core simulation engineering, and cloud or infrastructure teams.
Translate product and evaluation needs into model improvements including long-horizon prediction and scenario generation quality.
Lead and execute Gaia's post-training and closed-loop pipeline through fine-tuning and targeted data curation.
Push autoregressive generation towards longer, more stable rollouts and make the model deployment-ready.
Provide technical leadership through mentorship, code reviews, and setting high engineering and research standards.

Requirements

Hands-on experience with post-training or fine-tuning large-scale language, video, or other foundation models.
Experience with world models, autoregressive generation, and long-horizon generation.
Experience with diffusion or flow models alongside a solid understanding of 3D vision.
Strong understanding of model architecture to contribute to architectural and training decisions.
Strong hands-on engineering skills with modern machine learning stacks including PyTorch, debugging, and performance-minded development.
Minimum of 5 years of relevant industry experience.

Nice to Have

Experience with inference optimisation or deploying large models under latency and compute constraints.
Experience improving data and training pipelines and working across distributed training and infrastructure constraints.
Proven technical leadership through tech lead ownership, mentoring, or setting direction.

Skills and Tools

PyTorch
Large foundation models
World models
Diffusion and flow models
3D vision
Autoregressive generation

Salary and Benefits

Salaries benchmarked annually against the market
Meaningful equity sharing in the ownership and long-term success of the company
Relocation support and visa sponsorship where applicable
Hybrid working with core hours and access to vehicle workshops and labs
Learning and development budgets for training and conferences
Comprehensive benefits including health insurance, dental, enhanced parental leave, retirement or pension plans, access to therapists, wellbeing partnerships, and team socials

About the Company

Wayve builds the leading AI platform for autonomous driving, pioneering an end-to-end AI approach that enables vehicles to learn directly from real-world experience. The mapless and hardware-agnostic AI platform integrates with global OEM partners to scale automation from L2 plus to L4. The main hubs are located in London, Sunnyvale, Yokohama, Herzliya, Vancouver, and Leonberg.

How to Apply

Interested candidates should apply on the official company careers page.

Skills and tech stack

PyTorchWorld ModelsDiffusion ModelsFlow ModelsFoundation ModelsDiffusionFine-tuning

Questions candidates ask

Where is this job located?

This role is based in London, United Kingdom, and operates on a hybrid working model.

How many years of experience are required?

The role requires 5 plus years of relevant industry experience.

What is the primary focus of the GAIA team?

The team focuses on simulation evaluation and validation, working on Gaia, which is Wayve's world model trained on large-scale driving video.

What machine learning framework is explicitly mentioned in the requirements?

PyTorch is mentioned as part of the modern machine learning stack.

Is visa sponsorship available for this position?

Yes, relocation support and visa sponsorship are available where applicable.

Ready to apply?

The application is on the official wayve careers page.

Apply on company site