Staff Machine Learning Engineer (Gaia)

Join Wayve, a cutting-edge technology company focused on revolutionizing self-driving cars. As a Staff Machine Learning Engineer, you will lead the development of Gaia, our video world model, and have a significant impact on the future of autonomous driving. You will work closely with various teams, contribute to model architecture and training strategy, and provide technical leadership through mentorship and setting high engineering standards. Enjoy competitive compensation, private healthcare, paid time off, mental health resources, and opportunities for learning and development.

• Lead and execute large-scale training runs for video 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.
• In-depth experience training large-scale models (language, video, or other foundation models), including ownership of training at scale
• 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
• Strong understanding of model architecture and the ability to contribute meaningfully to architectural/training decisions
• Experience improving data/training pipelines and working across infrastructure constraints (distributed training, efficiency, reliability)
• 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
• Proven technical leadership (tech lead ownership, mentoring, setting direction across an area)
• Direct experience with world models, video generation, or long-horizon prediction