Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Staff ML Engineer, Gaia

£24.3 - £28.2 per hourEstimated
Full-time

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.

 

Vacancy posted 18 hours ago
Similar jobs that could be interesting for youBased on the Staff ML Engineer, Gaia in London vacancy
  • £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... 
    Suggested
    Hourly pay
    Full-time
    Hybrid working
    On-site
    Work from home

    Wayve

    London
    a month ago
  • £92k - £120k per annumEstimated
     ...retrieval, inference, evaluation, and more. We are looking for a strong engineer to join our team and help us build and scale our product in a...  ...working across backend, frontend, and interacting with LLMs and ML models. You will solve hard engineering problems in scalability... 
    Suggested
    Full-time

    Scale AI

    London
    a month ago
  • £87k - £115k 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... 
    Suggested
    Full-time
    Skilled worker visa

    recraft

    London
    18 hours ago
  • £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... 
    Suggested
    Hourly pay
    Full-time
    Hybrid working
    On-site
    Work from home

    Wayve

    London
    28 days ago
  • £111k - £147k per annumEstimated
     ...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 team is to run large transformer-based models efficiently on low-cost, low-power edge devices to enable... 
    Suggested
    Hourly pay
    Full-time
    Hybrid working
    On-site
    Work from home

    wayve

    London
    18 hours ago
  • £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... 
    On-site

    Graphcore

    London
    more than 2 months ago
  • £101k - £134k per annumEstimated
     ...started. There are a myriad of opportunities to use AI / ML as part of business processes in checkout, and we’re looking...  ...innovation at scale that matters. We’re looking for a Staff Level AL / ML engineer to accelerate our adoption into the AI era; helping us set our... 
    Hybrid working
    On-site

    checkout.com

    London
    18 hours ago
  • £93k - £123k per annumEstimated
     ...continuously improve the platform and our day to day workflows, rather than treating these as secondary projects. As a Staff Machine Learning Engineer, you will hold a hands-on technical position. You will be part of a team that builds, ships, and maintains the entire... 
    Hybrid working
    On-site
    Relocation

    moonpay

    London
    15 hours ago
  • £68k - £90k per annumEstimated
     ...intelligent systems that can evaluate, manage, and route content reliably at global scale. We’re seeking a Staff Machine Learning Engineer to build and scale foundational ML systems that power content understanding across Spotify. In this role, you’ll work on systems that... 
    Permanent

    spotify

    London
    more than 2 months ago
  • £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

    Waymo

    London
    more than 2 months ago
  • £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-time
    Hybrid working

    Waymo

    London
    more than 2 months ago
  • £107k - £140k per annumEstimated
     ...Description: We are looking for an independent, dedicated Staff Machine Learning Engineer to join the  Visual Computing Group (VCG) and contribute to...  ...and deploy advanced neural networks and machine learning (ML) for game rendering and streaming systems of SIE that exceed... 
    Hybrid working

    PlayStation Global

    London
    21 days ago
  • £155k - £163k per annum

     ...and training products. We are looking for researchers and software engineers who are passionate about developing machine learning techniques...  ...Experience solving complex production problems using state-of-the-art ML techniques ~ Experience taking research to production ~... 
    Full-time
    Hybrid working

    Waymo

    London
    a month ago
  • £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

    Waymo

    London
    more than 2 months ago
  • £150k per annum

     ...Senior Staff MLOps Engineer Location: London | Hybrid (3/4 days in office) We are looking for a Senior Staff MLOps Engineer to join a growing...  ...operates as it grows. The role is about taking innovative ML work and turning it into something robust, reliable and operational... 
    Permanent
    Full-time
    Hybrid working
    On-site
    Immediate start
    London
    2 days ago
  • £67k - £88k per annumEstimated
    We 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

    spotify

    London
    more than 2 months ago
  • £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

    circadiahealth

    Vauxhall, Greater London
    more than 2 months ago
  •  ...About the company A VC-backed AI/ML startup in West London building a novel foundation model for fully automated, unsupervised software delivery in embedded control systems. Early stage, high urgency, high transparency. They value directness over jargon and hands-on ownership... 
    Immediate start

    WhyHireWrong?

    London
    5 days ago
  • £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... 
    Permanent
    On-site

    Richard Wheeler Associates

    London
    more than 2 months ago
  • £103k - £135k per annumEstimated
     ...We're hiring a Senior ML Engineer to own the design, training, and deployment of a novel foundation model — from research through production, including the custom CUDA kernels that make it fast. This is a hands-on, high-ownership role for someone who has already shipped a... 
    Remote job
    Long-term contract
    Full-time
    Immediate start

    PulseRise Technologies LTD

    London
    12 days ago
  • £77k - £102k per annumEstimated
     ...progress. The defining work of the 21st century will be marked by the engine of inference delivering immense and diffuse chains of intellectual...  ...complex problems. The Role We are looking for Senior ML Compiler Engineers with experience in machine learning compilation... 
    On-site

    Fractile

    London
    4 days ago
  • £400 per day

    Company: OPUS RECRUITMENT SOLUTIONS Job Type: Contract, Full Time Salary: £0 - £400/day
    Full-time

    OPUS RECRUITMENT SOLUTIONS

    Finsbury, Greater London
    a month ago
  • £90k - £119k per annumEstimated
     ...Senior Machine Learning Engineer - Biomedical AI London - Hybrid (3 days a week in the office) We are partnering with an innovative, research-driven biotechnology company that is building next generation AI systems that help identify new therapeutic opportunities... 
    Permanent
    Full-time
    Hybrid working
    On-site
    London
    15 days ago
  • £83k - £109k 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... 
    On-site

    humanoid

    London
    18 hours ago
  • £53k - £70k per annumEstimated
     ...ML Runtime Engineer Location: Bristol / London  About Fractile Fractile was founded in 2022 on the bet that, eventually, the world’s most capable AI systems would be limited in their impact by the time taken to produce useful outputs. We bet everything on the logical... 
    On-site

    Fractile

    London
    5 days ago
  • £57k - £76k per annumEstimated
     ...global health. Now, we are looking for an experienced research engineer to supercharge our product. In this role, you will design and...  ...Requirements Industry experience in building and deploying ML systems in production. Strong focus on model validation and deployment... 
    Full-time
    Remote

    deepmirror

    London
    a month ago
  • £69k - £91k per annumEstimated
     ...rely on our best-in-class platform. Our Winning Culture is the engine that drives our teams of innovators. We champion diversity of thought...  ...build what’s next - together! Role Overview We are seeking a ML Ops Engineer to join our Platform Engineering team at Anaplan. In... 

    Anaplan

    London
    2 days ago
  • £100k - £150k per annum

     ...to this role , but to other relevant opportunities across its network, so one application can unlock multiple roles. Senior ML Engineer | Foundation Models & CUDA  TechTree's client is hiring a Senior ML Engineer to help build and scale a novel foundation model... 
    Full-time

    C&D Talent Advisory

    London
    3 days ago
  • £58k - £77k per annumEstimated
     ...people to help us scale. Who You Are You're a hands-on AI engineer with 3-5+ years of experience building production-grade...  ...Experience ~3-5+ years of hands-on experience building production AI/ML systems, with 1-2+ years deploying LLM applications to production... 
    Full-time
    Flexible hours

    AION

    London
    28 days ago
  • $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 job
    Summer work

    Mercor

    London
    16 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Staff ML Engineer, Gaia. Be the first to apply!