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

Staff ML Engineer, Gaia

Full-time

Wayve

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.

For more information visit Careers at Wayve.

To learn more about what drives us, visit Values at Wayve

For US candidates only, please visit E-Verify Notice and Participation and Right to Work

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.
Vacancy posted 1 day ago
Similar jobs that could be interesting for youBased on the Staff ML Engineer, Gaia in United Kingdom vacancy
  •  ...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... 
    Suggested
    Full time
    Work at office
    Remote work
    Relocation
    Home office
    Flexible hours
    2 days per week
    1 day per week

    Moonpay

    United Kingdom
    4 days ago
  •  ...scale, and optimise next-generation world model architectures (GAIA and successors) and bridge them into high-throughput...  ...same training stack we use for real driving. You sit between ML research and engineering: collaborating with scientists on architecture and conditioning... 
    Suggested
    Full time
    Work at office
    Work from home

    Wayve

    United Kingdom
    2 days ago
  •  ...remote, async. Nice to have Solana validator operations / staking infrastructure. MEV ecosystem — relayers, bundles, block engines. SUI, Hyperliquid, or other high-throughput chains. Real-time market-data / streaming systems at scale. Low-level... 
    Suggested
    Full time
    For contractors
    Remote work
    Overseas

    p2p.org

    United Kingdom
    20 days ago
  •  ...and one where everyone is valued. We strive for excellence in everything we do. The Role XTX is seeking an experienced engineer to support our ML Performance and AI Acceleration function. You will be working in a small and highly technical team at extremely high- and... 
    Suggested
    Full time
    Work at office
    Worldwide

    XTX Markets

    United Kingdom
    21 days ago
  •  ...outdoors and a desire to protect it for future generations.  Role Summary We are looking for a full-time Machine Learning Engineer, with deep knowledge and strong enthusiasm towards establishing a state-of-art AI infrastructure for training very large... 
    Suggested
    Full time
    Contract work
    Local area

    Rivian

    United Kingdom
    12 days ago
  •  ...businesses think about cybersecurity, digital experiences, and identity and access management. As a Ping Identity Staff Site Reliability Engineer, you will be involved in every facet of our Cloud-based services. You will establish solutions for building, deploying... 
    Full time
    Local area
    Worldwide
    Flexible hours

    Ping Identity

    United Kingdom
    22 days ago
  •  ...prediction models across ARQ’s countries and acquisition channels. Work closely with Growth & Marketing, Product, Finance, Data Engineering, and Engineering teams to understand business needs and translate them into modelling solutions. Help evaluate acquisition... 
    Full time
    Work at office
    3 days per week

    ARQ ( Prev DolarApp )

    United Kingdom
    22 days ago
  •  ...take problems from idea to production — partnering with Product, Engineering, and Operations teams to turn large-scale datasets into models...  .... Transform ambiguous business problems into measurable ML solutions. Partner closely with Product and Operations teams... 
    Full time
    Work at office
    3 days per week

    ARQ ( Prev DolarApp )

    United Kingdom
    22 days ago
  •  ...do is the result of the creativity and commitment of our team - builders doing the best work of their lives. We are researchers, engineers, and operators. IOI medalists and ex-founders. If you want to work hard and create lasting positive impact, we want to hear from you... 
    Full time
    Immediate start
    Remote work

    Valor Capital Group

    United Kingdom
    22 days ago
  •  ...understand the underlying behaviours employed by criminals. You will be product focused, working in close collaboration with our engineering and operations data scientists as well as the wider sales, consulting, and product teams.  In this position, you will: Perform... 
    Full time
    Worldwide

    Mastercard

    United Kingdom
    11 days ago
  •  ...to the team… Are pursuing a bachelor’s, master’s or PhD in mathematics, economics, physics, statistics, computer science or any engineering related field and have an expected graduation date between December 2027 and August 2028 Advanced quantitative, analytical... 
    Full time
    Work experience placement
    Summer work
    Internship
    Work at office
    Immediate start
    Day shift
    Afternoon shift

    Drw

    United Kingdom
    11 days ago
  •  ...partnership with Ingenico bringing stablecoin payments to 40M+ terminals across 120+ countries. The Role This is a Senior Data Engineer position on the Data Engineering team. You’ll work closely with the Data Engineering Lead to design and scale the data platform... 
    Full time
    Work at office
    Remote work
    Work from home
    Home office

    Reown

    United Kingdom
    11 days ago
  •  ...We're hiring a Staff QA Automation Engineer to help shape how quality engineering works across tem. QA sits inside Platform, a central team that builds the tooling and standards the rest of engineering runs on. You'll design the frameworks and practices that let them... 
    Full time
    Contract work

    tem

    United Kingdom
    1 day ago
  •  ...integrating a high-performance central limit order book matching engine with automated market making to provide deep and predictable...  ...+ years in backend development, with a proven track record in a Staff, Principal, or equivalent technical leadership role where you were... 
    Full time

    Bullish

    United Kingdom
    4 days ago
  •  ...-million ARR company with a team of 14 from AWS, Snyk, GitHub, Okta, Netlify, Uber, Samsara, and more. We are hiring a Staff Platform Engineer to own the infrastructure Nango runs on - and increasingly, the infrastructure our customers run Nango on. This is a critical... 
    Full time
    Remote work

    Nango

    United Kingdom
    2 days ago
  • $80 - $150 per hour

    Physics Expert (Postdoc / Junior professor) Pay: $80 - $150/hour Role Type: Contractor Location: Remote (US, Canada, UK focused) This remote contractor role engages Physics Experts at the postdoc or junior professor level for a high-impact project supporting...
    Remote job
    For contractors

    SaidGig

    United Kingdom
    2 days ago
  •  ...partnering across Credit Risk, Product, Engineering, Analytics, and Commercial teams to shape...  ...elegant models. Comfortable operating at staff level: you can set direction in ambiguity...  ...teams to deploy decisioning or ML systems into live products, rather than operating... 
    Full time
    Work at office
    Immediate start

    LemFi

    United Kingdom
    4 days ago
  •  ...experienced, technical, and hands-on Autonomy Data Manager to grow and lead a talented team composed of Data Scientists, Data Engineers and ML Ops Engineers, working in the Autonomy team to deliver data driven Planning and Perception solutions. You will lead the team to... 
    Full time
    Temporary work
    Work at office
    Flexible hours

    Tesla

    United Kingdom
    11 days ago
  •  ...reliable production workflows. ~ Python: You are comfortable using Python to build pipelines, automate processes, and develop data engineering tools. ~ Data modeling: You understand dimensional modeling, ELT patterns, and how to design datasets for reliability,... 
    Full time
    Remote work

    N8n

    United Kingdom
    24 days ago
  •  ...We are seeking a Senior Analytics Engineer to drive significant business impact by shaping our data infrastructure, developing robust data models, and promoting best practices in analytics engineering (e.g., data quality, documentation, and scalable data architectures... 
    Full time

    Paddle

    United Kingdom
    15 days ago
  •  ...enables consistent KPI definitions and supports downstream consumers, including LLM-based analytics agents. Collaborate with Data Engineers on upstream data contracts and event schemas - ensuring raw data is structured in a way that supports scalable, reliable... 
    Full time

    SoSafe

    United Kingdom
    2 days ago
  •  ...Your Next Destination We're looking for a Staff Data Scientist, Pricing to build the...  ...across the full lifecycle, from feature engineering and model selection through to deployment...  ...the decision. ~ Fluency in the Python ML stack alongside strong SQL , with the... 
    Full time

    Airalo

    United Kingdom
    11 days ago
  • $1,000 per month

     ...Do you want to help define the future of finance? Are you a strong software engineer who's passionate about data, distributed systems, and delivering impact at scale? Are you looking for a values-driven company that invests in its people and gives you the autonomy to... 
    Remote job
    Full time
    Home office

    Elliptic

    United Kingdom
    19 days ago
  • From empowering entrepreneurs to build successful businesses, to supporting their clients to arrange 'me time' moments, we’re in the business of elevating people - helping them to thrive and feel fantastic. Lifting others up can be hard, heavy work, and we passionately ...
    Full time

    Booksy

    United Kingdom
    24 days ago
  •  ...to grow your career at Booksy. The people you’ll like to work with and things you'll enjoy impacting: As an Associate Data Engineering Manager reporting to the Head of Data Engineering in our Data & Analytics - Data Engineering team, your purpose will be to lead a... 
    Full time
    Work at office
    Remote work

    Booksy

    United Kingdom
    a month ago
  •  ...via Claude, return accurate, contextually aware results. You are the steward of how well our BI tool thinks. Partner with data engineering to optimise performance, and build what's missing Own and maintain the semantic data layer and transformation logic so that... 
    Full time

    Primer

    United Kingdom
    24 days ago

Do you want to receive more vacancies?

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