Staff Software Engineer, AI Libraries

📋 Description

• Work closely with machine learning teams to understand their workflows and turn complex
• Spend much of your time designing and writing Python, leading projects from discovery through
• Improve engineering standards and support users as they push the scale and performance of their
• Build modular libraries for areas such as multimodal data loading, distributed training and
• Improve the reliability, performance and developer experience of systems used across Wayve, helping
• Set technical direction, and raise the bar for software architecture, testing and observability.

🎯 Requirements

• Deep Python expertise
• Proven track record of leading the design, delivery and long-term evolution of high-quality
• Ability to 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.
• Experience with concurrent, parallel or distributed computing.
• Familiarity with ML frameworks such as PyTorch or TensorFlow.

🎁 Benefits

• Salaries benchmarked against the market annually
• Meaningful equity, sharing in the ownership and long term success of Wayve
• Relocation support and visa sponsorship where applicable
• Hybrid working, core hours and the chance to work hands on in vehicle workshops and labs
• Learning and development budgets with support for training, conferences and growth
• Comprehensive benefits including health insurance, dental, enhanced maternity and paternity leave