Staff Software Engineer, AI Libraries

Overview

As a Staff Software Engineer on Wayve’s AI Platform team, you design and evolve reusable libraries and platforms that enable ML teams to train, evaluate and deploy models at scale. You’ll shape APIs and architectures that support multimodal data, distributed training and large GPU clusters, while improving reliability and developer experience. You collaborate with ML researchers to translate complex requirements into scalable software that accelerates autonomous-driving advances. This role offers ownership, impact, and the chance to influence how Wayve learns from real-world data.

Pay / Benefits

• salary benchmarked against market
• equity/ownership
• relocation support and visa sponsorship
• hybrid working with core hours
• learning and development budgets
• comprehensive benefits including health insurance and enhanced parental leave

Responsibilities

• Design and implement modular libraries for multimodal data loading, distributed training and inference, and observability.
• Improve system reliability, performance and developer experience across Wayve’s AI stack.
• Lead end-to-end projects from discovery through adoption and maintenance, guiding other engineers.
• Translate ML team requirements into stable, reusable APIs and architectures.
• Collaborate with cross-functional teams to scale model training on large GPU clusters.

Key requirements

• Deep Python expertise and track record of leading design and delivery of large-scale software systems.
• Experience turning complex ML requirements into reusable libraries and clear APIs.
• Ability to set technical direction and raise the bar on architecture, testing and observability.
• Hands-on leadership while driving projects end-to-end.

• Collaborative
• Impact-driven
• Mentoring/leadership
• Python
• distributed computing
• PyTorch or TensorFlow

…