b Overview /b p Join Wayve to optimize AI workloads for autonomous driving. You’ll drive performance improvements across ML training and cloud inference, enabling larger models and faster deployment. You’ll collaborate with Research and model teams to embed performance into development cycles and influence cloud GPU strategies. Your work will scale efficiency across targets with reusable optimisations and benchmarking. This role offers hands-on impact in a fast-moving, mission-driven environment. /p b Pay / Benefits /b ul li hybrid working model /li li learning and development budgets /li li meaningful equity /li li relocation support and visa sponsorship /li li health insurance and wellbeing offerings /li li enhanced maternity/paternity leave /li /ul b Responsibilities /b ul li Profile and optimize ML workloads for training and cloud inference to reduce bottlenecks /li li Design and implement cross-target optimisations (kernels, data loaders, frameworks like Triton) /li li Develop benchmarking tools to quantify efficiency gains and detect regressions /li li Collaborate with Research, model teams, and platform teams to align performance with product goals /li li Inform cloud GPU hardware strategy and readiness /li li Foster a culture of performance optimization across teams /li /ul b Key requirements /b ul li 10+ years in performance engineering for ML systems or GPU compute infrastructure /li li Experience with large-scale GPU workloads (training or inference) /li li Proven ability to write, report and track performance benchmarks /li li High-quality, well-structured Python code /li li BS or MS in Machine Learning, Computer Science, Engineering or related field, or equivalent experience /li /ul ul li cross-functional collaboration /li li strong communication of benchmarks and results /li li context-driven problem solving /li li GPU compute infrastructure /li li profiling tools and kernel-level profiling /li li CUDA or Triton GPU kernel development (nice to have) /li /ul
Staff ML Performance Engineer, Performance Architecture employer: wayve
Wayve is an exceptional employer, offering a dynamic and inclusive work culture that fosters innovation and collaboration. With a focus on employee growth, you will have the opportunity to work alongside talented AI engineers in a hybrid environment, contributing to groundbreaking advancements in autonomous driving while enjoying the vibrant atmosphere of London. Join us to make a meaningful impact in the field of AI and data science.
