Wayve
Staff ML Performance Engineer, Performance Architecture
$370K–$395K
Sunnyvale, California USA · Hybrid · Staff · Full-time · Posted Oct 2, 2026
- Experience
- 10+ years of industry experience.
- Education
- BS or MS in Machine Learning, Computer Science, Engineering, or a related technical discipline, or equivalent experience.
- Visa sponsorship
- Visa sponsorship where applicable.
About the job
Optimizes machine-learning training and cloud-inference performance through workload profiling, reusable improvements, benchmarking, and collaboration with research and platform teams.
What you'll do
- Profile workloads and identify bottlenecks; implement and benchmark performance optimizations.
- Build reusable, cross-target optimizations and tools for training and cloud inference.
- Advise on cloud GPU hardware strategy and promote performance optimization with research and model teams.
What they ask for
- 10+ years of industry experience in performance engineering across ML systems, GPU compute infrastructure, distributed platforms, or similar.
- Experience optimizing large-scale workloads on GPU compute clusters for training, inference, or both.
- Experience writing, reporting, and tracking performance benchmarks accessibly.
- High-quality, well-structured, tested Python code.
Nice to have
- Experience with concurrent, parallel, and distributed computing; inference-serving optimization; system profilers such as NVIDIA Nsight Systems; or GPU kernels such as CUDA or Triton.
- Knowledge of computing fundamentals, including what makes code fast, secure, and reliable.
Pay
$370,040 – $394,900 (total pay; offers equity)
Tools
Python, Triton, NVIDIA Nsight Systems, CUDA
AI focus
Improves the performance and efficiency of ML training and cloud inference workloads.
Model-written summary · the original posting is authoritative