Principal Machine Learning Engineer GAIA

Home > Python programming jobs

Principal Machine Learning Engineer GAIA in UK

Wayve
Full time
Email

UK
09/25/2026
Report

London

Responsibilities

Lead Gaia’s post-training and closed-loop pipeline, including fine-tuning, alignment, experimentation, and targeted data curation.
Improve long-horizon autoregressive generation so Gaia produces stable, coherent, and steerable rollouts.
Contribute to model architecture and training-strategy decisions across pre-training, post-training, and application layers.
Translate post-training improvements into measurable downstream impact with research, applications, simulation engineering, and cloud/infrastructure teams.
Provide technical leadership through mentorship, reviews, and setting high engineering and research standards.
Improve inference optimization, deployment readiness, reliability, and performance under relevant constraints.

Requirements

Hands-on experience 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 and an understanding of 3D vision.
Strong understanding of model architecture and ability to contribute to architecture and training decisions.
Strong hands-on engineering skills with modern ML stacks, including PyTorch, debugging, and performance- and reliability-minded development.
Typically 5+ years of relevant industry experience.
Advanced degrees are valued, but substantial applied experience is important.
Experience with inference optimization, large-model deployment under latency or compute constraints, data or training pipelines, distributed training, efficiency, reliability, and technical leadership is desirable.

Benefits

Full-time position based in London, UK, with a hybrid working arrangement combining office/workshop and home working.
Core working hours with flexibility to determine a schedule that works for the individual and team.
Inclusive interview experience with accommodations or adjustments available on request.

Wayve

Wayve builds end-to-end autonomous driving software—the vehicle-agnostic Wayve AI Driver—that runs on onboard compute and native sensors, licensed to automakers and fleet operators. Its platform spans ADAS and higher autonomy (L2+/L3 to robotaxi) and is designed to generalize across vehicle types...

Similar positions

Graduate Software Engineer

Synack
Full time

UK
10/04/2026

Remote

Software Engineer - Cloud Images

Canonical
Full time

Remote
10/04/2026

EMEA, Americas

GTM Engineer

Higgsfield
Full time

USA
10/04/2026
Salary: $130k - $175k/yr

San Francisco, CA

Research Engineer

Clera
Full time

Others
10/04/2026

Zürich, Switzerland

Lead Automation Quality Engineer

London Stock Exchange Group
Full time

Others
10/04/2026
Salary: Competitive

Bengaluru, India