Software Engineer AI Libraries
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 design, build, and maintain Python libraries and tools for training, evaluating, and scaling models. You will develop ML workflow abstractions, support distributed training, optimise data pipelines, and improve the reliability, observability, testing, and usability of training infrastructure.
Requirements
- Python
- Software system design and delivery experience
- Software architecture
- Testing
- Observability
- Maintainability
- Cloud environment experience
- Azure
- Concurrent, parallel, or distributed computing
- PyTorch, TensorFlow, or PyTorch Lightning
- Technical stakeholder collaboration
Responsibilities
- Design, build, and maintain scalable Python libraries and tools
- Develop abstractions for data loading, distributed training, inference, checkpointing, and model evaluation
- Support training across GPU clusters and cloud infrastructure
- Create reliable, documented, observable, and adoptable tools
- Improve architecture, testing, monitoring, and maintainability across ML systems
- Optimise data and training pipelines for multimodal sensor data
- Contribute to AI-platform evolution