Lead large-scale post-training and fine-tuning of Gaia, Wayve's world model for autonomous driving simulation. Partner with research and engineering teams to improve model architecture, long-horizon prediction, and deployment reliability. Requires 5+ years of experience in foundation model training, world models, and modern ML stacks like PyTorch.
Key Highlights
Own and lead large-scale training and post-training pipelines for the Gaia world model.
Contribute to model architecture decisions and improve long-horizon autoregressive generation.
Provide technical leadership, mentorship, and set engineering standards across the simulation team.
Key Responsibilities
Own and lead parts of large-scale training for Gaia, from experimentation through productionised training runs.
Contribute to and influence model architecture decisions for the world model.
Partner across sub-teams including Research, Applications, Core Simulation Engineering, and Cloud/Infrastructure.
Translate product and evaluation needs into model improvements such as better long-horizon prediction and scenario generation quality.
Mentor team members and set technical direction to raise engineering and research quality standards.
Lead and execute Gaia's post-training and closed-loop pipeline, including fine-tuning and data curation.
Push Gaia's autoregressive generation towards longer, more stable rollouts and ensure deployment readiness.
Technical Skills Required
PyTorch
World Models
Model Fine-tuning
Discover our full range of relocation jobs with comprehensive support packages to help you relocate and settle in your new location.
Benefits & Perks
Market-benchmarked salaries
Meaningful equity
Relocation support and visa sponsorship
Hybrid working model
Learning and development budgets
Comprehensive health and dental insurance
Enhanced maternity and paternity leave
Retirement or pension contributions
Nice to Have
Experience with inference optimisation or deploying large models under latency/compute constraints.
Experience improving data/training pipelines and working across infrastructure constraints.
Proven technical leadership experience such as tech lead ownership or mentoring.
Job Description
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.
Want the full job description?
Read the complete details on LinkedIn, the original posting.
Continue on LinkedIn
This is a short excerpt. All rights to the full description belong to its original publisher.
