Staff AI Platform Engineer — Libraries & Scale (Hybrid)

Wayve; Sunnyvale (CA); Hybrid; USD 190.000 - 270.000; Vollzeit

Benefits dieser Stelle

Equity

Relocation & Visa

Hybrid work

Learning budget

Benefits package

Zusammenfassung

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

Qualifikationen

• 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.

Aufgaben

• 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.

Kenntnisse

Python

Distributed systems

PyTorch

TensorFlow

Azure cloud

Leadership

Jobbeschreibung

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