Python AI Platform Engineer - Libraries & GPUs

Wayve; City Of London; Hybrid; GBP 90,000 - 130,000; Full time

Benefits offered by this job

Equity / stock options

Relocation support & visa sponsorship

Hybrid working with office + remote

Learning & development budget

Job summary

Wayve is seeking a skilled software/platform engineer to help build scalable Python-based libraries and tools for its AI Platform team. You will design, implement and maintain systems used by researchers and engineers for training, evaluating and deploying models at scale.

You will work closely with ML teams to translate complex requirements into robust software, emphasizing architecture, observability and performance.

Qualifications

• Strong Python experience and track record delivering scalable software systems.
• Experience with software architecture, testing and observability.
• Familiarity with concurrent, parallel or distributed computing and ML frameworks such as PyTorch or TensorFlow.
• Experience with cloud environments, ideally Azure.

Responsibilities

• Design and build modular libraries and platforms for ML engineers.
• Lead projects from discovery through delivery, improving engineering standards.
• Collaborate with ML teams to understand workflows and scale models on large GPU clusters.

Skills

Python

Distributed systems

PyTorch

TensorFlow

Azure

Software architecture

Testing and observability

Tools

PyTorch

TensorFlow

Azure

Job description

Wayve is seeking a skilled software/platform engineer to help build scalable Python-based libraries and tools for its AI Platform team. You will design, implement and maintain systems used by researchers and engineers for training, evaluating and deploying models at scale.

You will work closely with ML teams to translate complex requirements into robust software, emphasizing architecture, observability and performance.