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
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
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.
