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 …
Skills: Large-scale Model Training, PyTorch, Model Architecture, Video Generation, World Models
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 …
Skills: Large-scale Model Training, PyTorch, Model Architecture, Video Generation, Foundation Models
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 …
Skills: Technical Program Management, Data Platform, Data Pipelines, People Leadership, Distributed Data Platforms
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 …
Skills: Program Delivery, Technical Program Management, Systems Thinking, Risk Management, Cross-functional Collaboration
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 …
Skills: Program Management, Stakeholder Management, Data Operations, Vendor Management, Risk Management
Staff ML Performance Engineer (Inference Optimisation)
London, England, United Kingdom · Hybrid
Senior$2.5B raised
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 …
Skills: ML Inference Optimisation, CUDA, TensorRT, Quantisation, Kernel Development
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 …
Skills: Wiring Harness Construction, Hardware Troubleshooting, System Integration, HIL Setups, Sensor Calibration
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 …
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 …
Skills: Distributed Systems, GPU Cloud Infrastructure, AI Training, Inference Workloads, Petabyte-Scale Data Architecture
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 …
Skills: Vehicle Diagnostics, Fault Triage, Fleet Operations, ADAS Technology, Process Development
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 …
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 …
Skills: Technical Program Management, Software Development Lifecycle, Agile Methodologies, Systems Engineering, Project Planning
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 …
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 …
Skills: Program Delivery, Technical Program Management, Systems Thinking, Risk Management, Cross-functional Collaboration
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 …
Skills: Technical Program Management, Software Development Lifecycle, Agile Methodologies, Systems Engineering, Project Planning
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 …
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 …
Skills: Technical Program Management, Data Platform, Data Pipelines, People Leadership, Distributed Data Platforms
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 …
Skills: Program Management, Stakeholder Management, Data Operations, Vendor Management, Risk Management
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 …
Skills: Distributed Systems, ML Ops, System Architecture, Kubernetes, Spark
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 …
Sign up with Clera and we'll reach out the moment a role actually fits you — no more spraying applications into the void.
Full-time
postgraduate degree
Posted 1d ago
~40 hrs/week
Responsibilities
Lead the training and improvement of frontier-scale video world models to generate synthetic driving scenarios. Provide technical leadership through mentorship and the establishment of high engineering and research standards.
Requirements
Requires 4-5+ years of industry experience training large-scale foundation models with strong PyTorch skills. Depth of applied experience in model architecture and distributed training is essential.
Full job description
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
Gaia is Wayve’s video world model: trained on large-scale driving video, it predicts future frames from past context—functioning as a simulator that helps generate synthetic scenarios, including rare or safety-critical events. As a Staff ML Engineer on Gaia, you’ll own and drive work on training and improving frontier-scale models trained in-house. This is a high-impact role with the opportunity to tech-lead a key area and help shape the next version of Gaia in a fast-paced, results-focused environment.
Key responsibilities:
Lead and execute large-scale training runs for video (or adjacent) foundation models, from experimental design through production-grade execution
Contribute to model architecture and training strategy, using first-principles understanding rather than “off-the-shelf” application
Improve world-model capabilities that enable synthetic scenario generation and downstream evaluation/training of the driving model
Partner closely with research, applications, simulation engineering, and cloud/infrastructure teams to deliver end-to-end impact
Provide technical leadership through mentorship, review, and setting high engineering/research standards (Senior/Staff scope)
About you
In order to set you up for success as a Staff ML Engineer (Gaia) at Wayve, we’re looking for the following skills and experience.
Essential
In-depth experience training large-scale models (language, video, or other foundation models), including ownership of training at scale
Strong understanding of model architecture and the ability to contribute meaningfully to architectural/training decisions
Strong hands-on engineering skills with modern ML stacks (e.g., PyTorch), including debugging and performance/reliability-minded development
Relevant industry experience (typically 4–5+ years); advanced degrees are valued, but depth of applied experience is important
Desirable
Direct experience with world models, video generation, or long-horizon prediction
Experience improving data/training pipelines and working across infrastructure constraints (distributed training, efficiency, reliability)
Proven technical leadership (tech lead ownership, mentoring, setting direction across an area)
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.
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.
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.
At Wayve, we’re building a global driving intelligence that learns from data and scales across different vehicles and geographies.
Founded in 2017, we have pioneered an end-to-end AI approach to autonomous driving that is faster to deploy, more flexible by design and built to scale.
We deliver all levels of autonomy, from hands-off and eyes-off driving, to robotaxis. We license and integrate the Wayve AI Driver as a vehicle-agnostic software platform that runs entirely on onboard vehicle compute and native sensors.
Wayve is the first and only AV company to test a single global AI Driver model across more than 500 cities in Europe, North America and Japan.
We’re building autonomy for anyone, in any vehicle, anywhere.
Offices: ., London, United Kingdom, GB · 709 N Shoreline Blvd, Mountain View, California 94043, US
At Wayve, we’re building a global driving intelligence that learns from data and scales across different vehicles and geographies.
Founded in 2017, we have pioneered an end-to-end AI approach to autonomous driving that is faster to deploy, more flexible by design and built to scale.
We deliver all levels of autonomy, from hands-off and eyes-off driving, to robotaxis. We license and integrate the Wayve AI Driver as a vehicle-agnostic software platform that runs entirely on onboard vehicle compute and native sensors.
Wayve is the first and only AV company to test a single global AI Driver model across more than 500 cities in Europe, North America and Japan.
We’re building autonomy for anyone, in any vehicle, anywhere.
Offices: ., London, United Kingdom, GB · 709 N Shoreline Blvd, Mountain View, California 94043, US