Principal Machine Learning Engineer GAIA

Before the detail, here's the challenge you'd help us solve. We build the embodied intelligence that moves real vehicles safely, and the ecosystem a billion machines will run on in the future. Very few people in AI can say this. Every role here, whatever the team, plugs into that. Here’s what this particular role covers.

You'll be joining the Simulation team (Simulation Evaluation & Validation), working on Gaia — Wayve's world model. Gaia is trained on large-scale driving video to predict future frames from past context, functioning as a simulator that generates synthetic scenarios and operates in closed loop with the driving model itself.

The team partners closely with Research, Applications, Core Simulation Engineering, and Cloud/Infrastructure to turn model improvements into measurable downstream impact on the driving stack. Your day-to-day - Own and lead parts of large-scale training for Gaia (language- or video-style training at scale), from experimentation through productionised training runs. - Contribute to and influence model architecture decisions — not just applying existing models. - Partner across sub-teams: Research (model development), Applications (adapting models to use cases), Core Simulation Engineering, and Cloud/Infrastructure. - Translate product and evaluation needs into model improvements (e.g. better long-horizon prediction, scenario generation quality, failure-case coverage).