AI Platform Engineer — Scalable ML Libraries

Wayve; Greater London; On-site; GBP 80,000 - 120,000; Full time

Job summary

Wayve is seeking a Software Engineer to join the AI Libraries team. You will build and maintain Python libraries and tools that empower Wayve’s ML engineers and researchers to train, evaluate, and scale models efficiently.

This hands-on role focuses on stable, scalable, and modular systems enabling large-scale ML development, collaborating with ML teams to design reusable abstractions and improve reliability and performance of training infrastructure.

Qualifications

• Strong Python programming experience and ability to design, build, and maintain software systems end-to-end.
• Strong software architecture and system design skills with focus on scalable libraries and tooling.
• Experience working in cloud environments, ideally Azure.
• Experience with concurrent, parallel, or distributed computing.
• Familiarity with ML frameworks such as PyTorch and TensorFlow.
• Ability to collaborate with technical stakeholders to refine requirements and deliver practical 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 evaluation workflows.
• Support training at scale across large GPU clusters and cloud-based infrastructure.
• Collaborate with ML teams to understand user needs and create reliable, well-documented tools.
• Improve engineering quality across ML systems through architecture, testing, monitoring, and maintainability practices.
• Optimize data and training pipelines for multi-modal data sources (camera, radar, lidar, etc.).
• Contribute to the evolution of Wayve's AI platform as we scale autonomous driving capabilities.

Skills

Python

System design

Azure

Distributed computing

PyTorch

TensorFlow

PyTorch Lightning

Stakeholder communication

Tools

Docker

Kubernetes

Terraform

Prometheus

Grafana

Datadog

OpenTelemetry

Airflow

Flyte

Ray

Metaflow

Argo Workflows

NVIDIA Nsight

Job description

Wayve is seeking a Software Engineer to join the AI Libraries team. You will build and maintain Python libraries and tools that empower Wayve’s ML engineers and researchers to train, evaluate, and scale models efficiently.

This hands-on role focuses on stable, scalable, and modular systems enabling large-scale ML development, collaborating with ML teams to design reusable abstractions and improve reliability and performance of training infrastructure.