Software Engineer, AI Libraries
Salary not listed • London, United Kingdom • Full-time
Posted 26d ago
Remote Work Policy
On-site
Employment Type
Full-time
Categories
Machine Learning Engineer
About the job
Wayve is building the leading AI platform for autonomous driving, pioneering an end-to-end AI approach that enables vehicles to learn directly from real-world experience. This role is within the AI Platform team, specifically on the AI Libraries team, where you will create high-quality platforms, tools, and reusable abstractions to support thousands of GPUs and accelerate the development of autonomous driving technology. You will work closely with machine learning teams to understand their workflows and translate complex requirements into stable, scalable, and intuitive software, focusing on designing and writing Python, leading projects from discovery to delivery, and improving engineering standards.
Responsibilities
- Design and write Python code to build modular libraries for areas like multimodal data loading, distributed training, inference, checkpointing, and observability.
- Improve the reliability, performance, and developer experience of systems used across the company.
- Help teams train models efficiently on large GPU clusters and work with various data sources (camera, radar, lidar).
- Understand machine learning team workflows and translate complex requirements into stable, scalable, and intuitive software.
- Lead projects from discovery through delivery and improve engineering standards.
- Support users as they push the scale and performance of their models.
Requirements
- Extensive Python experience.
- Track record of designing, delivering, and maintaining high-quality software systems at scale.
- Understanding of software architecture, testing, and observability.
- Experience with concurrent, parallel, or distributed computing.
- Familiarity with ML frameworks such as PyTorch or TensorFlow.
- Familiarity with cloud environments, ideally Azure.
- This is a software and platform engineering role, not an ML modeling position.
Benefits
- Salaries benchmarked against the market annually
- Meaningful equity
- Relocation support
- Visa sponsorship where applicable
- Hybrid working
- Core hours
- Chance to work hands-on in vehicle workshops and labs
- Learning and development budgets
- Support for training and conferences
- Comprehensive benefits including health insurance, dental, enhanced maternity and paternity leave, retirement or pension, access to therapists, wellbeing partnerships, team socials