Staff Software Engineer, AI Libraries

Overview

In this Staff-level role, you will architect and deliver scalable, reusable software platforms that empower Wayve’s ML teams to train and deploy models at scale. You’ll shape API design and libraries used across thousands of GPUs, enabling efficient work with multimodal data and large-scale training. You’ll lead end-to-end projects, mentor peers, and set technical direction to raise the bar in reliability, performance, and observability. This is a hands-on, ownership-driven position in a fast-evolving AI platform team.

Pay / Benefits

• competitive salary
• meaningful equity
• relocation support and visa sponsorship
• hybrid working model
• learning and development budget
• comprehensive benefits (health, dental, Enhanced leave, retirement)

Responsibilities

• Design and deliver stable, scalable libraries and APIs for ML workflows
• Lead projects from discovery to adoption and maintenance
• Improve reliability, performance, and developer experience of platform systems
• Build modular libraries for multimodal data loading, distributed training/inference, checkpointing, and observability
• Collaborate with ML teams to translate requirements into reusable software across large GPU clusters
• Guide and mentor engineers, driving end-to-end delivery and architectural standards

Key requirements

• Deep Python expertise with a proven track record of leading design and delivery of high-quality software systems at scale
• Ability to transform complex ML requirements into reusable libraries and clear APIs
• Set technical direction and raise standards for architecture, testing, and observability
• Hands-on leadership while guiding others and delivering projects end-to-end

• Collaborative mindset
• Strong communication and mentoring abilities
• Strategic thinking with hands-on execution
• Python
• Concurrent/parallel/distributed computing
• PyTorch or TensorFlow

…