W

Senior Performance Engineer, AI Infrastructure

wayve • United State
Visa Sponsorship Relocation
Apply Now

Optimize and profile large-scale ML training and cloud inference workloads to maximize efficiency and throughput. Design reusable cross-target optimizations and benchmarking tools for GPU compute clusters. Requires 10+ years of experience in performance engineering, Python, and GPU infrastructure.

Key Highlights
10+ years of industry experience in ML systems performance engineering
Focus on optimizing training and cloud inference workloads on GPU clusters
Hybrid working model with hubs in London, Sunnyvale, Yokohama, Herzliya, Vancouver, and Leonberg
Key Responsibilities
Identify, quantify, and deliver optimizations across training and cloud inference workloads
Profile workloads to find bottlenecks using system and kernel level profilers
Design and implement efficiency improvements to maximize MFU, throughput, and utilization
Build reusable, cross-target optimizations such as kernels, data loaders, and frameworks
Design and implement benchmarking tools to track efficiency gains and catch regressions
Inform cloud GPU hardware strategy and readiness in partnership with platform teams
Build a culture of performance optimization with Research and model teams
Technical Skills Required
Python GPU Compute Infrastructure Performance Profiling
Benefits & Perks
Market-benchmarked salaries
Meaningful equity
Relocation support and visa sponsorship where applicable
Hybrid working with core hours
Learning and development budgets
Comprehensive benefits including health insurance, dental, and enhanced parental leave
Nice to Have
Experience with concurrent, parallel, and distributed computing
Experience optimizing inference serving systems (latency, throughput, batching, caching)
Experience using NVIDIA Nsight Systems or other system profilers
Experience implementing GPU kernels (CUDA, Triton, etc.)
Knowledge of computing fundamentals regarding speed, security, and reliability

Job Description

Before the detail, here's the challenge you'd help us solve.We build the embodied intelligence that moves real vehicles safely, and the ecosystem a billion machines will run on in the future. Very few people in AI can say this. Every role here, whatever the team, plugs into that.Here’s what this particular role covers.The Performance Architecture team is part of Wayve's AI Performance org. We make Wayve's AI workloads faster and more efficient across training and cloud inference, so that performance unlocks new product capability. Our work lets Wayve train larger models faster and run inference more efficiently at scale.
Want the full job description? Read the complete details on LinkedIn, the original posting.
Continue on LinkedIn

This is a short excerpt. All rights to the full description belong to its original publisher.

See Jaabz jobs first on Google 1 tap · free · in AI Overviews Jaabz is on your Google Manage Preferred Sources

Similar Jobs

Explore other opportunities that match your interests

Visa Sponsorship Relocation Remote
Job Type Full-time
Experience Level Not Applicable

wayve

United State

Senior Software Engineer - Cloud Security (Golang/GCP)

Programming
•
3m ago

Premium Job

Sign up is free! Login or Sign up to view full details.

•••••• •••••• ••••••
Job Type ••••••
Experience Level ••••••

Palo Alto Networks

United State

Executive Director, Claims Data & Analytics

Programming
•
5m ago

Premium Job

Sign up is free! Login or Sign up to view full details.

•••••• •••••• ••••••
Job Type ••••••
Experience Level ••••••

USAA

United State

Subscribe our newsletter

New Things Will Always Update Regularly