Software Engineer, AI Libraries

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. 🛠️ About our AI Platform Team 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. 🧠 Your day-to-day You’ll work closely with machine learning teams to understand their workflows and turn complex requirements into stable, scalable and intuitive software. You’ll spend much of your time designing and writing Python, leading projects from discovery through delivery, improving engineering standards and supporting users as they push the scale and performance of their models. 🧩 What you’ll be working on You’ll build modular libraries for areas such as multimodal data loading, distributed training and inference, checkpointing and observability. You’ll improve the reliability, performance and developer experience of systems used across Wayve, helping teams train models efficiently on large GPU clusters and work with camera, radar, lidar and other data sources. 🙌 You…