Principal Machine Learning Engineer GAIA
Wayve · London, United Kingdom
Roboticsremoteprincipal
At a glance
Highlights
- Lead Gaia post-training to improve long-horizon prediction and scenario generation.
- Build a closed-loop simulator that generates synthetic scenarios for driving models.
- Shape embodied intelligence that moves real vehicles safely at scale.
What they offer
- Salaries benchmarked against the market annually
- Meaningful equity, sharing in ownership and long-term success
- Relocation support and visa sponsorship where applicable
- Hybrid working with core hours and hands-on vehicle workshops
- Learning and development budgets for training and conferences
- Comprehensive benefits including health insurance, dental and parental leave
Ideal candidate
Experienced applied ML leader passionate about autonomy who thrives with real ownership in ambiguous settings and cares about turning research into safer real-world driving.
Requirements
- Hands-on experience post-training/fine-tuning large-scale models
- Experience with world models, autoregressive and long-horizon generation
- Experience with diffusion/flow models and solid understanding of 3D vision
- Strong understanding of model architecture and training decisions
- Strong hands-on engineering skills with modern ML stacks (e.g. PyTorch)
- 5+ years of relevant industry experience
- Experience with inference optimisation or deploying large models under constraints
- Experience improving data/training pipelines across infrastructure constraints
- Proven technical leadership - ownership, mentoring and setting direction
- remote policy
- Hybrid working with core hours
- visa sponsorship
- Visa sponsorship and relocation support where applicable
Lab profile
About Wayve
London-based AI lab building embodied driving intelligence for autonomous vehicles.
What they build
- Wayve Driver embodied AI driving software
- GAIA generative world models for driving
- LINGO language-grounded driving AI
Known for
- End-to-end embodied AI for driving
- Camera-first, mapless autonomy approach
- Large-scale real-world learning and simulation
Key facts
- Founded in 2017
- Headquartered in London, UK
- Backed by SoftBank Group, NVIDIA and Microsoft
Posted · Last seen