Software Engineer, AI Libraries & Platform

Wayve; London (KY); On-site; USD 120,000 - 160,000; Full time

Job summary

Wayve is seeking a Software Engineer to join its AI Libraries team. You will build and maintain Python libraries and tools that empower ML engineers to train, evaluate, and scale models. The role emphasizes robust abstractions, reliable tooling, and scalable infrastructure across GPU clusters and cloud environments.

You will collaborate with ML teams to understand needs, improve observability, and advance Wayve’s AI platform as autonomous driving capabilities scale for real-world deployment.

Qualifications

• Strong Python programming experience and ability to design, build, and deliver software.
• Strong architecture and system design skills for scalable tools and libraries.
• Experience with cloud environments, ideally Azure.
• Experience with concurrent, parallel, or distributed computing.
• Familiarity with ML frameworks such as PyTorch, TensorFlow, or PyTorch Lightning.
• Ability to refine requirements with technical stakeholders to deliver scalable solutions.

Responsibilities

• Design, build, and maintain scalable Python libraries and tools used by ML engineers and researchers across Wayve.
• Develop robust abstractions for data loading, distributed training, inference, checkpointing, and model evaluation workflows.
• Support training at scale across large GPU clusters and cloud-based infrastructure.
• Work closely with ML teams to understand user needs and create tools that are reliable, well-documented, observable, and easy to adopt.
• Improve engineering quality across ML systems through strong software architecture, testing, monitoring, and maintainability practices.
• Optimise data and training pipelines to support multi-modal data sources, including camera, radar, lidar, and other sensor data.
• Contribute to the evolution of Wayve’s AI platform as we scale our autonomous driving capabilities.

Skills

Python programming

Software architecture

Cloud environments (Azure)

Distributed computing

ML frameworks (PyTorch, TensorFlow)

Stakeholder collaboration

Tools

Airflow

Docker

Kubernetes

Observability tools

Job description

Wayve is seeking a Software Engineer to join its AI Libraries team. You will build and maintain Python libraries and tools that empower ML engineers to train, evaluate, and scale models. The role emphasizes robust abstractions, reliable tooling, and scalable infrastructure across GPU clusters and cloud environments.

You will collaborate with ML teams to understand needs, improve observability, and advance Wayve’s AI platform as autonomous driving capabilities scale for real-world deployment.