Staff ML 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 large-scale training runs for video and foundation models, from experiment design through production execution. You will contribute to model architecture and training strategy, improve world-model capabilities, and work across research, application, simulation, and infrastructure disciplines. You will mentor others and set high technical standards.
Requirements
- Large-scale model training
- Foundation model
- Model architecture
- PyTorch
- Machine learning engineering
- Debugging
- Distributed training
- Training efficiency
- Training reliability
- World model
- Video generation
- Long-horizon prediction
- Technical leadership
- Mentoring
Responsibilities
- Lead large-scale model training runs
- Contribute to model architecture and training strategy
- Improve world-model capabilities for scenario generation and driving-model evaluation
- Partner across research, application, simulation, and infrastructure functions
- Provide technical leadership, mentorship, and review
Benefits
- Hybrid working policy