Staff Software Engineer, AI Libraries

Wayve • Sunnyvale, CA

10d • $276,100 - $311,400 • Hybrid

About The Position

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.

Requirements

• Deep Python expertise.
• Proven track record of leading the design, delivery and long-term evolution of high-quality software systems at scale.
• Ability to turn complex requirements across ML teams into stable, reusable libraries and clear APIs.
• Ability to set technical direction.
• Ability to raise the bar for software architecture, testing and observability.
• 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.
• This is a Staff-level software and platform engineering role, not an ML modelling position.

Nice To Haves

• Familiarity with ML frameworks such as PyTorch or TensorFlow.
• Familiarity with cloud environments—ideally Azure.

Responsibilities

• Work closely with machine learning teams to understand their workflows and turn complex requirements into stable, scalable and intuitive software.
• Design and write Python code.
• Lead projects from discovery through delivery.
• Improve engineering standards.
• Support users as they push the scale and performance of their models.
• Build modular libraries for areas such as multimodal data loading, distributed training and inference, checkpointing and observability.
• Improve the reliability, performance and developer experience of systems used across Wayve.
• Help teams train models efficiently on large GPU clusters and work with camera, radar, lidar and other data sources.

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, retirement or pension where applicable, access to therapists, wellbeing partnerships, team socials and more

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: No Education Listed