Senior ML Performance Engineer — Scale Training & Inference (Hybrid)

Wayve; Sunnyvale (CA); Hybrid; USD 180,000 - 250,000; Full time

Benefits offered by this job

Salaries benchmarked annually

Equity

Relocation support

Hybrid working

Learning & development budgets

Comprehensive benefits

Job summary

Wayve is building the leading AI platform for autonomous driving with a mapless, hardware-agnostic approach. This role focuses on performance engineering across ML systems, GPU compute and distributed platforms, shaping optimisations for training and cloud inference.

You will profile workloads, implement cross-target improvements, collaborate with Research, model and platform teams, and deliver benchmarks and tooling to track gains while advancing Wayve's AI driver toward real-world deployment.

Qualifications

• 10+ years of industry experience driving performance engineering across ML systems, GPU compute infrastructure, and distributed platforms.
• Optimised large-scale workloads on GPU compute clusters for training and inference.
• Written, reported and tracked performance benchmarks in an open, accessible way.
• Proficiency in Python with high-quality, well-structured, tested code.
• BS or MS in Machine Learning, Computer Science, Engineering or related technical discipline, or equivalent experience.

Responsibilities

• Profile ML workloads to identify bottlenecks across training and cloud inference.
• Design and implement cross-target optimisations (kernels, data loaders, frameworks like Triton).
• Build reusable benchmarking tools to track gains and catch regressions.
• Collaborate with Research, model, and platform teams to integrate performance into development.
• Inform cloud GPU hardware strategy and readiness with platform teams.
• Foster a culture of performance optimisation within teams.

Skills

Performance engineering

Python

GPU compute

Profiling

System optimization

Education

BS or MS in ML/CS/Engineering or related

Tools

Nsight Systems

CUDA

Triton

Job description

Wayve is building the leading AI platform for autonomous driving with a mapless, hardware-agnostic approach. This role focuses on performance engineering across ML systems, GPU compute and distributed platforms, shaping optimisations for training and cloud inference.

You will profile workloads, implement cross-target improvements, collaborate with Research, model and platform teams, and deliver benchmarks and tooling to track gains while advancing Wayve's AI driver toward real-world deployment.