Python Platform Engineer: AI Libraries for Scalable GPUs

Wayve; Greater London; Hybrid; GBP 110,000 - 140,000; Full time

Benefits offered by this job

Meaningful equity

Relocation support

Hybrid working

Learning and development budget

Health insurance

Dental insurance

Pension

Job summary

Wayve is seeking an experienced Software Platform Engineer to join the AI Platform Team in London. You will design and implement modular libraries, mainly in Python, to support training, evaluation and deployment of models at scale across GPU clusters.

You will collaborate with ML researchers and engineers, focusing on reliability, performance and observability. The role is hybrid, with in-person collaboration and remote work options.

Qualifications

• Extensive Python experience and ability to design, deliver and maintain large-scale software systems.
• Experience with software architecture, testing and observability.
• Familiarity with ML frameworks such as PyTorch or TensorFlow and cloud environments (Azure) is valuable.

Responsibilities

• Design and write Python, lead projects from discovery through delivery, and improve engineering standards.
• Build modular libraries for data loading, training, inference, and observability.
• Support users and help teams scale model training on large GPU clusters.

Skills

Python

Distributed computing

Observability

Tools

PyTorch

TensorFlow

Azure

Job description

Wayve is seeking an experienced Software Platform Engineer to join the AI Platform Team in London. You will design and implement modular libraries, mainly in Python, to support training, evaluation and deployment of models at scale across GPU clusters.

You will collaborate with ML researchers and engineers, focusing on reliability, performance and observability. The role is hybrid, with in-person collaboration and remote work options.