> Markdown version of [/jobs/ext/4541904-staff-software-engineer-ai-libraries](https://www.wearedevelopers.com/jobs/ext/4541904-staff-software-engineer-ai-libraries). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# Staff Software Engineer, AI Libraries

- **Company:** Wayve Technologies Ltd
- **Location:** Sunnyvale, CA, United States (Remote available)
- **Contract:** Permanent contract
- **Skills:** LTE (Telecommunication), Application Programming Interfaces (APIs), Artificial Intelligence, Microsoft Azure, Computer Clusters, Extract Transform Load (ETL), Distributed Computing Environment, Distributed Systems, Python (Programming Language), Machine Learning, Software Architecture, Tensorflow, Software Engineering, Software Systems, Scripting, Graphics Processing Unit (GPU), Pytorch, AI Platforms, Machine Learning Operations, Lidar
- **Published:** November 3, 2026
- **Apply:** https://www.careerbuilder.com/job-details/staff-software-engineer-ai-libraries-sunnyvale-ca--d46020d9-2786-49bd-b12e-b7bd144e1950

## About the Role

Software Engineering, Artificial Intelligence (AI), Ecosystems, Artificial Intelligence (AI), Machine Learning, Training/Teaching, GPU (Graphics Processing Unit), Autonomous Driving Systems, Python Programming/Scripting Language, Performance Modeling, You have deep Python expertise and a proven track record of leading the design, delivery and long-term evolution of high-quality software systems at scale. You can turn complex requirements across ML teams into stable, reusable libraries and clear APIs, set technical direction, and raise the bar for software architecture, testing and observability. You remain hands-on while guiding other engineers and driving projects end to end, from shaping requirements through to adoption and maintenance. Experience with concurrent, parallel or distributed computing is important; familiarity with ML frameworks such as PyTorch or TensorFlow and cloud environments-ideally Azure-is highly valuable. This is a Staff-level software and platform engineering role, not an ML modelling position.

Not ticking every box? That's totally okay! If you're passionate about autonomy and keen to learn, we encourage you to apply even if you don't meet every requirement.

## Description

Before the detail, heres the challenge youd 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.