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.
- £67k - £89k per annumEstimated...Wayve the experience that defines your career! The role Gaia is Wayve’s video world model: trained on large-scale driving video... ...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...SuggestedHourly payFull-timeHybrid workingOn-siteWork from home
- £105k - £136k per annumEstimated...Back to jobs tags.new Sr Staff ML Forward Deployed Engineer, Enterprise GenAI London, UK Scale GP (Scale Generative AI Platform) is an enterprise-grade Generative AI platform that provides APIs for knowledge retrieval, inference, evaluation, and more. We are looking...SuggestedFull-time
- £70k - £92k per annumEstimated...environment; we back each other to deliver impact. Make Wayve the experience that defines your career! The role As a Staff ML Performance Engineer, you’ll play a key role in high-impact projects, optimising ML inference for edge accelerators and GPUs. The focus of this...SuggestedHourly payFull-timeHybrid workingOn-siteWork from home
- £77k - £101k per annumEstimated...define the future, come and join the team ! Where do you want to work? What do you want to do? 117 jobs open AI Platform AV Engineering Commercial & Operations Fleet Operations Hardware & Supply Chain People & Talent Product & Delivery Product &...SuggestedFull-timeShift work
- £55k - £74k per annumEstimated...About the job Validate the ML stack that turns accelerator hardware into trusted AI performance. This role sits where modern ML... ...distributed workloads, simulators, emulators and hardware targets. Engineers are expected to take ownership, question assumptions and improve...SuggestedOn-site
£115k - £135k per annum
...conditions, while Accumail finally allows staff-to-staff communication to happen instantly... ...Team We are a mission-driven team of 80 engineers based in London and the surrounding areas,... ...communication platform. As a Staff ML/AI Engineer, you'll operate across multiple...Long-term contractFull-timeOn-siteImmediate startFlexible hours£155k - £163k per annum
...up the Waymo driver. We are looking for researchers and software engineers who are passionate about developing machine learning techniques for... ...-on experience in machine learning projects ~ Experience with ML frameworks such as TensorFlow, PyTorch, Hugging Face's...Full-time£155k - £163k per annum
...several evaluation and training products. We are looking for research engineers to work on these exciting problems. In this hybrid role, you... ...solving complex production problems using state-of-the-art ML techniques ~ Experience taking research to production ~ Expertise...Full-timeHybrid working- £77k - £102k per annumEstimated...systems that can evaluate, manage, and route content reliably at global scale. We’re seeking a Senior Staff Machine Learning Engineer to build and scale foundational ML systems that power content understanding, safety, and decisioning across the platform. In this role, you...Permanent
£150k - £162k per annum
...tens of billions in simulation across 15+ U.S. states. The DUE ML Core London team builds and operates scalable machine learning... ...the Waymo Driver. We are looking for researchers and software engineers passionate about developing ML techniques for evaluation systems...Full-time£155k - £163k per annum
...billions in simulation across 15+ U.S. states. The Simulation ML Infrastructure team builds scalable AI/ML infrastructure to accelerate... ...We seek an experienced Senior Machine Learning Infrastructure Engineer to lead the development of advanced AI/ML infrastructure for...Full-time- £67k - £88k per annumEstimatedWe design Spotify’s consumer experience—end to end, moment to moment, across every screen, platform, and partner integration. Our mission is to make listening feel effortless, personal, and joyful for billions of users around the world. That means turning complexity into clarity...Permanent
- £72k - £96k per annumEstimated...within Computational Innovation to enhance and accelerate portfolio decision-making. THE POSITION We are looking for a Senior ML Engineer to join the AI Systems team and work at the frontier of biomedical AI. This is a hands-on engineering role with real stakes as the...Long-term contractFull-timeHybrid workingOn-site
- £68k - £90k per annumEstimated...Position Overview As an ML Ops Engineer at Circadia Health , you will own the infrastructure and operational lifecycle of the machine learning systems that power our clinical monitoring platform. You will build and maintain the production ML pipelines, deployment infrastructure...Full-time
£90k - £120k per annum
Machine Learning Engineer – Hedge Fund Forecasting Central London – office based £90-120k highly negotiable + discretionary bonus 50-100%... ...Research teams to own the design, implementation, and maintenance of ML pipelines for forecasting. You will prize taking ownership of the...PermanentOn-site- £77k - £102k per annumEstimated...the cutting edge of the latest AI developments in both hardware and software. Want to get involved? We are looking for Senior ML Compiler Engineers with experience in machine learning compilation to work on the compiler stack for our ground-breaking AI accelerators. You can...On-site
- £76k - £99k per annumEstimated...even further. About The Role We're hiring a Senior Dev/MLOps Engineer to join our Data & Compute Platform team based in London. What You’ll Do Own and manage the full lifecycle of both ML models and core infrastructure - from development and deployment to monitoring...Full-timeOn-site
- £66k - £85k per annumEstimated...passionate about pushing the boundaries of AI, we want you on board! Job Description We are looking for experienced Research Engineers to develop and optimise AI models, conduct cutting-edge research, and push the boundaries of AI and machine learning. Whether your...Full-timeSkilled worker visa
- £53k - £70k per annumEstimated...ML Runtime Engineer (Mid-Level and Senior) Fractile · London or Bristol · Full-time · Hybrid About Fractile We’re taking a revolutionary approach to computing — building AI acceleration hardware that runs the world’s largest language models 100× faster than existing...Full-timeHybrid workingOn-site
- £95k - £126k per annumEstimated..., healthier lives. Role Purpose As a Lead Machine Learning Engineer focused on MLOps within the Data Science Platform team, you will... ...sensors and devices, you will design and operate production-grade ML systems, ensuring strong experiment tracking, model lifecycle management...Full-timeHybrid workingOn-siteFlexible hours
- £56k - £73k per annumEstimated...Description At Recraft, we’re building the next generation of generative models across images and text. We’re looking for an ML Data Engineer to scale our data pipelines for unstructured data (primarily images) and keep our training flows fast, reliable, and repeatable....Full-timeSkilled worker visa
- £48k - £63k per annumEstimated...Research Engineering (Machine Learning) We are looking for Research Engineers with different... ...levels of experience - Mid through to Senior, Staff, Principal or equivalent levels. We are... ...the performance and robustness of ML models, using a full spectrum of state-of-...Hybrid workingOn-site1 day/week
- £92k - £122k per annumEstimated...Position Overview As an ML Research Engineer at Circadia Health , you will research, design, and build the next generation of models and algorithms that power our clinical monitoring platform. Circadia's devices use radar to continuously and contactlessly capture respiratory...Full-time
$85 per hour
...include Benchmark , General Catalyst , Peter Thiel , Adam D'Angelo , Larry Summers , and Jack Dorsey . Position: ML Engineer (Coding Agent Experience) Type: Contract Compensation: $85/hour Location: Remote Role Responsibilities...Remote jobSummer workMercor
London16 days ago$75 - $90 per hour
...General Catalyst , Peter Thiel , Adam D'Angelo , Larry Summers , and Jack Dorsey . Position: Human Baseliner for Open-Ended ML Research Tasks Type: Contract Compensation: $75–$90/hour Location: Remote Commitment: 20+ hours/...Remote jobSummer workMercor
London8 days ago- ...We are looking for a Senior ML Systems Engineer to build and validate simulation infrastructure for large-scale machine learning systems. This role focuses on modelling the compute and communication behaviour of systems used for ML training and inference, and using simulation...
- £75k - £100k per annumEstimated...your impact and unlock incredible career growth opportunities, join us, and build real world value. THE WORK: We're looking for a Staff Data Scientist to be the technical lead across Ripple's diverse product and business portfolio. You'll define the analytics vision,...Full-timeOn-site
- £74k - £99k per annumEstimated...a wholly-owned subsidiary of Sony Group Corporation. As a Staff Data Scientist , you will lead innovation in experimentation and... ...technical stakeholders. ~ Experience working with modern data engineering and visualization tools (e.g., Airflow, Git, Tableau, MicroStrategy...Hybrid working
- £35k - £46k per annumEstimated...cutting edge of Machine Learning and AI. About the Role To strengthen our AI for Science (AI4S) team, we are looking for AI/ML Engineers with a track record in developing and validating machine learning models for real-world scientific problems. You will drive the development...Long-term contractFull-timeHybrid workingRemote
- £76k - £102k per annumEstimated...Most engineers get to use large language models. You will get to build the systems that run them in production, at scale, under real... ...is expanding its AI Core group and is looking for a Senior or Staff AI & ML Engineer in the UK. This is hands-on, build-focused work at the...Full-timeRemote
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Staff ML Engineer, Gaia. Be the first to apply!
- ai machine learning engineer London
- graduate machine learning engineer London
- machine learning software engineer London
- ai ml engineer remote London
- machine learning engineer London
- machine learning engineer remote London
- talent acquisition consultant (machine learning) London
- data scientist machine learning London
- intern quantum machine learning for quantum computing London
- machine learning London