Principal Machine Learning Engineer GAIA
Wayve is a London-headquartered embodied-AI company developing and licensing mapless, vehicle-agnostic driving software for assisted, automated, and robotaxi applications.
About Wayve
Wayve Technologies Ltd. develops the Wayve AI Driver, an end-to-end, data-trained software platform that runs on onboard vehicle compute and native sensors. It is designed for OEM integration across L1 driver assistance through L4 automated driving, without HD maps.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will lead Gaia’s post-training and closed-loop pipeline, fine-tune and align world models, and curate targeted data. You will improve autoregressive rollout stability, steerability, inference performance, and reliability. You will contribute to architecture and training decisions, collaborate across technical functions, and provide technical leadership through mentorship and reviews.
Requirements
- Hands-on experience post-training or fine-tuning large-scale foundation models
- Experience with world models, autoregressive generation, and long-horizon generation
- Experience with diffusion or flow models
- Understanding of 3D vision
- Strong understanding of model architecture and training decisions
- Strong engineering skills with modern ML stacks such as PyTorch
- Relevant industry experience, typically 5+ years
Responsibilities
- Lead and execute Gaia's post-training and closed-loop pipeline
- Fine-tune and align the world model through experimentation and targeted data curation
- Improve autoregressive generation for longer and more stable rollouts
- Make the model deployment-ready, including inference performance and reliability
- Contribute to model architecture and training-strategy decisions
- Translate post-training improvements into measurable downstream impact
- Provide technical leadership through mentorship, review, and high engineering and research standards
Benefits
- Hybrid working policy with office and home working
- Core working hours with flexibility to determine your schedule