Staff AI Platform Engineer — Libraries & Scale (Hybrid)

Wayve; Sunnyvale (CA); Hybrid; USD 190,000 - 270,000; Full time

Benefits offered by this job

Equity

Relocation & Visa

Hybrid work

Learning budget

Benefits package

Job summary

Wayve is building the core AI platform for autonomous driving. We are seeking aStaff-level software and platform engineer to design, develop and evolve high‑quality libraries that support ML teams at scale.

You will lead end‑to‑end projects, shaping requirements and defining architectural direction in a hands‑on role. You’ll work closely with ML engineers and researchers, focusing on data loading, distributed training, checkpointing and observability across large GPU clusters, while advocating

Qualifications

• Deep Python expertise and ability to lead design and delivery of scalable software systems.
• Experience turning ML workflows into stable, reusable libraries and clear APIs.
• Strong background in concurrent, parallel, or distributed computing.

Responsibilities

• Design, write and lead Python-based software for ML platforms and libraries.
• Turn complex ML requirements into stable, reusable APIs and libraries.
• Guide projects end-to-end from discovery through adoption and maintenance.
• Collaborate with ML teams to improve reliability, performance and observability.

Skills

Python

Distributed systems

PyTorch

TensorFlow

Azure cloud

Leadership

Job description

Wayve is building the core AI platform for autonomous driving. We are seeking aStaff-level software and platform engineer to design, develop and evolve high‑quality libraries that support ML teams at scale.

You will lead end‑to‑end projects, shaping requirements and defining architectural direction in a hands‑on role. You’ll work closely with ML engineers and researchers, focusing on data loading, distributed training, checkpointing and observability across large GPU clusters, while advocating