How your CV stacks up
Upload your CV to see how well it fits this job role
?%
Staff ML Engineer, Gaia
Wayve: Staff ML Engineer – Gaia (Hybrid, London-based)
About Us
Founded in 2017, Wayve is the leading developer of Embodied AI technology. Our advanced AI software and foundation models empower vehicles to perceive, understand, and navigate complex environments, enhancing safety and usability in automated driving.
Our vision is to pioneer autonomy that advances the world. With mapless, hardware-agnostic solutions for automakers, we accelerate the progression from assisted to fully automated driving.
Thriving in an unpredictable, fast-paced environment, we tackle complexity head-on to deliver transformative solutions. Our culture values bold ambitions paired with humility—continuous growth and collaboration drive us toward a smarter, safer future.
At Wayve, your contributions shape progress. We celebrate diversity, champion new perspectives, and foster inclusivity, ensuring every team member’s impact resonates.
The Role
Staff ML Engineer (Gaia)
Gaia is Wayve’s video world model—trained on large-scale driving video, it predicts future frames from context, acting as a simulator to generate synthetic scenarios (including rare or critical events). As the tech lead, you’ll own end-to-end training initiatives for this frontier-scale in-house model.
This high-impact position blends innovation, scalability, and technical leadership. You’ll collaborate cross-functionally to advance Gaia’s capabilities and drive Wayve’s long-term goals within a results-driven, adaptive environment.
Reasons to use Rodeo
Start with a chat, not a search bar
Grad scheme, placement, apprenticeship? Not sure what you want yet — that's fine. Your agent talks it through with you and turns "I have no idea" into a shortlist.
It searches the market for you
Every day your agent scans the market matching roles against what actually matters to you, not just keywords on a CV.
Only hits
No noise. No "maybe this fits." Just roles with a clear explanation of why they're right — and where to focus when applying.
Key Responsibilities
- Lead large-scale training pipelines for video/adjacent foundation models, from experimental design to production execution
- Shape model architecture and training strategies with first-principles thinking rather than broadly applicable frameworks
- Enhance world-model capabilities to improve synthetic scenario generation and downstream evaluation/training of the driving model
- Partner strategically with research, applications, simulation, and cloud/infrastructure teams for end-to-end impact
- Provide technical leadership through:
- Mentorship of senior engineers
- Setting rigorous ML/research standards
- Designing high-quality guidelines and documentation
About You
We’re seeking a Staff ML Engineer with exceptional expertise in large-scale model training and passion for shaping the future of autonomy.
Essential
- 8+ years of hands-on ML experience (орд advanced degrees, prior deep technical experience redeems academic gaps)
- Ownership of training pipelines for foundation models (essential)
- Fluency in model architecture design with the ability to innovate beyond off-the-shelf solutions
- Practical ML engineering skills with modern stacks:
- Proficiency in PyTorch
- Experience debugging complex systems,
- Prioritising reliability and scalability in distributed/tight infrastructure
Get help with your application
Your very own career expert that helps elevate your application to the next level.
Desirable
- Direct experience with world models, video generation, or long-horizon prediction
- Ability to optimise data/training pipelines under constraints (distributed systems, efficiency, resilience)
- Proven technical leadership:
- Leading cross-functional projects,
- Mentoring engineers,
- Aligning teams around technical direction
Work Environment
Location: Primary base in our London office, with a hybrid working model balancing creative collaboration in-person and remote efficiency.
We ensure flexible, responsible hiring practices:
- Disability/DEI accommodations are discussed openly throughout the process.
- All protected characteristics are not deliberated in interviews.
- Optional DEI monitoring form collects voluntary demographic data to improve inclusivity.
At Wayve, we take pride in an culture equally defined by:
- Curiosity and first-principles problem-solving
- Inclusivity and fair opportunity
- Impact and synergistic growth
Further Information
- Career opportunities: Wayve Careers
- Core values: Wayve Values
- Legal notices (US candidates only):
“It took my CV and asked me questions relevant to understanding what kind of jobs to suggest for me. Suggestions were almost perfect. Jobs were exactly what I’ve been looking for.”
Jessica, London
Skills
Location