Staff Software Engineer, AI Libraries
$276k - $311k • Sunnyvale, California USA • Full-time
Posted 17d ago
Job Location
Sunnyvale, California USA
Tech Stack
Remote Work Policy
On-site
Employment Type
Full-time
Categories
LLM Engineer
About the job
Wayve is building the leading AI platform for autonomous driving, pioneering an end-to-end AI approach that enables vehicles to learn directly from real-world experience. This role is on the AI Libraries team, which builds the foundations that enable Wayve's machine learning engineers and researchers to train, evaluate, and deploy models at scale. You will create high-quality platforms, tools, and reusable abstractions that support thousands of GPUs and accelerate the development of autonomous-driving technology.
You will work closely with machine learning teams to understand their workflows and translate complex requirements into stable, scalable, and intuitive software. Your day-to-day will involve designing and writing Python, leading projects from discovery through delivery, improving engineering standards, and supporting users as they push the scale and performance of their models. You will build modular libraries for areas such as multimodal data loading, distributed training and inference, checkpointing, and observability, improving the reliability, performance, and developer experience of systems used across Wayve.
Responsibilities
- Design and write Python code for AI platforms and tools.
- Lead projects from discovery through delivery.
- Improve engineering standards and support users of AI systems.
- Build modular libraries for data loading, distributed training, inference, checkpointing, and observability.
- Improve the reliability, performance, and developer experience of AI systems.
- Translate complex machine learning team requirements into stable, reusable libraries and clear APIs.
- Set technical direction and raise the bar for software architecture, testing, and observability.
- Remain hands-on while guiding other engineers and driving projects end-to-end.
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 translate complex requirements across ML teams into stable, reusable libraries and clear APIs.
- Experience setting technical direction and improving software architecture, testing, and observability.
- Experience with concurrent, parallel, or distributed computing.
- Familiarity with ML frameworks such as PyTorch or TensorFlow.
- Familiarity with cloud environments, ideally Azure.
Benefits
- Salaries benchmarked against the market annually
- Meaningful equity
- Relocation support
- Visa sponsorship
- Hybrid working
- Core hours
- Chance to work hands-on in vehicle workshops and labs
- Learning and development budgets
- Support for training, conferences, and growth
- Comprehensive benefits including health insurance, dental, enhanced maternity and paternity leave, retirement or pension, access to therapists, wellbeing partnerships, team socials