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 reliable, self-service performance tools that scale 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 level.
  • Predict latency, memory, utilization, and compute cost for model or recipe changes before compute is spent.
  • Own monitoring and regression alerting across model builds and training runs.
  • Collaborate with training and runtime engineers to establish performance targets and influence priorities with data.

Requirements

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

Benefits

  • Full-time role with a hybrid working policy combining time in Wayve offices and workshops with time working from home.
  • Inclusive interview accommodations and adjustments are available upon request.

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

Research Scientist - Machine Learning

  • ISEE
  • Full time
  • USA
  • 09/13/2026
  • Remote

Senior Software Engineer, ML Ops

  • PathAI
  • Full time
  • USA
  • 09/13/2026
  • Salary: $128k - $196k/yr
  • Boston, MA

Senior Machine Learning Engineer

  • Entefy
  • Full time
  • USA
  • 09/13/2026
  • Remote

Senior Machine Learning Engineer - Fraud

  • Plaid
  • Full time
  • USA
  • 09/13/2026
  • Salary: $229k - $315k/yr
  • Seattle, WA

Machine Learning Engineer

  • Entefy
  • Full time
  • USA
  • 09/13/2026
  • Remote

Listings

by


Job Type

Language

Country

CodingJobBoard.com © 2026