Senior Software Engineer, AI Platform & Libraries (Python, Distributed Systems)

wayve • United Kingdom

Visa Sponsorship Relocation

Apply Now

AI Summary

Design and build scalable Python-based AI libraries and platforms for Wayve’s autonomous driving systems, optimizing distributed training, multimodal data pipelines, and model deployment across GPU clusters. Collaborate with ML teams to enhance developer experience, reliability, and performance for large-scale autonomous vehicle AI models. Requires deep expertise in software architecture, distributed computing, and ML frameworks like PyTorch/TensorFlow.

Key Highlights

Build modular, high-performance AI libraries for distributed training, multimodal data loading, and observability

Optimize systems for large-scale GPU clusters and real-world autonomous driving data (camera, radar, lidar)

Work closely with ML engineers to improve developer experience and scalability of autonomous vehicle AI models

Looking to advance your Development & Programming career with relocation support? Explore Development & Programming Jobs with Relocation Packages that include comprehensive packages to help you move and settle in your new role.

Key Responsibilities

Design and implement high-quality, reusable Python libraries for distributed AI training and inference

Collaborate with ML teams to translate complex workflows into scalable, stable, and intuitive software solutions

Optimize systems for performance, reliability, and developer experience across large GPU clusters

Develop tools for multimodal data loading (camera, radar, lidar) and observability frameworks

Improve engineering standards, testing, and concurrent/parallel computing capabilities for autonomous driving AI

Technical Skills Required

Python Distributed Computing PyTorch or TensorFlow

Discover our full range of relocation jobs with comprehensive support packages to help you relocate and settle in your new location.

Benefits & Perks

Market-benchmarked annual salaries

Meaningful equity and long-term ownership

Hybrid working with core hours and access to vehicle workshops/labs

Interested in relocating to United Kingdom? Check out our comprehensive Relocation Jobs in United Kingdom page with detailed relocation packages and benefits.

Nice to Have

Experience with Azure cloud environments

Familiarity with ML frameworks beyond PyTorch/TensorFlow (e.g., JAX, TensorFlow Extended)

Background in autonomous systems or robotics

Job Description

Before the detail, here's the challenge you'd help us solve.We build the embodied intelligence that moves real vehicles safely, and the ecosystem a billion machines will run on in the future. Very few people in AI can say this. Every role here, whatever the team, plugs into that.Here’s what this particular role covers.Our AI Platform teams build the foundations that enable Wayve’s machine learning engineers and researchers to train, evaluate and deploy models at scale. Within the AI Libraries team, you’ll create high-quality platforms, tools and reusable abstractions that support thousands of GPUs and help accelerate the development of our autonomous-driving technology.

Want the full job description? Read the complete details on LinkedIn, the original posting.

Continue on LinkedIn

This is a short excerpt. All rights to the full description belong to its original publisher.