Your browser does not support javascript! Please enable it, otherwise web will not work for you.

Machine Learning Engineer, Performance Tooling

Home > Python programming jobs

Machine Learning Engineer, Performance Tooling in UK new

  • London

Responsibilities

  • Design and build scalable, self-service AI performance tools across models, hardware targets, and development workflows.
  • Define standards for measuring and predicting AI performance across the organization.
  • Model theoretical platform peaks, compare them with achieved performance, and identify efficiency losses at the layer and operator levels.
  • Predict model latency, memory, utilization, and compute cost before running expensive changes.
  • Own monitoring and regression alerting for model builds and training runs.
  • Collaborate with training and runtime engineers to establish performance targets and use data to influence priorities.

Requirements

  • Deep hands-on experience in performance engineering for complex systems, including profiling, roofline analysis, latency and throughput optimization, and workload bottleneck analysis.
  • Experience owning a tool or service end to end, from design and delivery through adoption by other teams.
  • Strong Python skills and experience profiling and instrumenting large production codebases.
  • Hands-on experience developing deep learning models with PyTorch.
  • Strong data analysis skills and the ability to derive defensible conclusions from noisy measurements.
  • Ability to turn ambiguous performance questions into measurable problems and prioritize impactful work.
  • Clear quantitative communication skills for influencing other teams’ priorities.

Benefits

  • Full-time role based in-office with a hybrid working policy combining office and workshop time with work from home.
  • Inclusive interview experience with accommodations or adjustments available upon request.
  • Inclusive and respectful workplace committed to diversity and equal treatment.

Wayve

Wayve builds end-to-end autonomous driving software—the vehicle-agnostic Wayve AI Driver—that runs on onboard compute and native sensors, licensed to automakers and fleet operators. Its platform spans ADAS and higher autonomy (L2+/L3 to robotaxi) and is designed to generalize across vehicle types...

Similar positions

Senior Software Engineer, Add-Ons

  • Mozilla
  • Full time
  • Canada
  • 09/16/2026
  • Salary: C$94k-C$139k
  • Remote

Senior Staff Machine Learning Engineer

  • Reddit
  • Full time
  • USA
  • 09/16/2026
  • Salary: $292k-$410k
  • Remote

Senior Machine Learning Engineer

  • NewsBreak
  • Full time
  • USA
  • 09/16/2026
  • Salary: $185k - $235k/yr
  • Bellevue, WA

Staff Software Engineer, Data Platform

  • Beacon Biosignals
  • Full time
  • USA
  • 09/16/2026
  • Salary: $170k - $190k/yr
  • Remote

Senior Software Engineer, Add-Ons

  • Mozilla
  • Full time
  • Remote
  • 09/16/2026
  • Salary: €58k-€77k
  • Spain

Listings

by


Job Type

Language

Country

CodingJobBoard.com © 2026