Wayve
Staff Software Engineer, AI Libraries
$276K–$311K
Sunnyvale, California USA · Staff · Full-time · Posted Oct 14, 2026
- Visa sponsorship
- Visa sponsorship where applicable.
About the job
Builds and leads scalable Python libraries and platform tools that help machine-learning teams train, evaluate, and deploy autonomous-driving models across large GPU clusters.
What you'll do
- Design and deliver reusable libraries for data loading, distributed training and inference, checkpointing, and observability.
- Improve the reliability, performance, and developer experience of systems used across Wayve.
- Lead projects end to end, set technical direction, and support machine-learning teams and other engineers.
What they ask for
- Deep Python expertise and a track record leading the design, delivery, and evolution of high-quality software systems at scale.
- Ability to turn complex requirements into stable, reusable libraries and clear APIs, and guide software architecture, testing, and observability.
- Experience with concurrent, parallel, or distributed computing.
- Hands-on technical leadership, including guiding engineers and driving projects through adoption and maintenance.
Nice to have
- Familiarity with ML frameworks such as PyTorch or TensorFlow.
- Familiarity with cloud environments, ideally Azure.
Pay
$276.1K – $311.4K (USD, base salary)
Tools
Python, PyTorch, TensorFlow, Azure
AI focus
Builds AI platform libraries and tools supporting the training, evaluation, and deployment of autonomous-driving models; this is not an ML modelling position.
Model-written summary · the original posting is authoritative