Staff ML Engineer Gaia
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
