Machine Learning Engineer, Performance Tooling

Salary: £60,000 - 60,000 per year

Requirements:

• Deep, hands-on performance engineering in complex systems, including profiling, roofline analysis, latency and throughput optimization, and root-causing workload limits.
• Proven experience owning a tool or service end to end, from design and delivery through to adoption by other teams.
• Strong Python skills, with comfort profiling and instrumenting large production codebases.
• Hands-on experience developing deep learning models with PyTorch.
• Data analysis skills to turn noisy measurements into defensible conclusions.
• Strong judgment to turn ambiguous performance questions into measurable ones and prioritize what matters.
• Quantitative communication skills clear enough to influence another teams priorities.

Responsibilities:

• Design and build reliable, self-service performance tools that scale across models, hardware targets, and development workflows.
• Shape how we measure and predict AI performance, and set standards other teams build on.
• Model theoretical peak for a platform, compare it with achieved performance, and identify where efficiency is lost at layer and op level.
• Predict latency, memory, utilization, and compute cost of model or recipe changes before compute is spent.
• Own monitoring and regression alerting across model builds and training runs.
• Work with training and runtime engineers to set performance targets and make the case with data.
• Build tools that reason across the AI stack, including models and operators, compilers, runtimes, accelerators, and distributed training infrastructure.
• Turn profiling data into a clear picture of where time, memory, power, and compute go, and what proposed changes will do to latency, throughput, compute spend, and capacity.
• Collaborate closely with model, compiler, runtime, platform, and hardware teams to turn measurement into clear recommendations.

Technologies:

• AI
• Cloud
• Embedded
• Hardware
• Model Training
• PyTorch
• Python

More:

Founded in 2017, we are the leading developer of Embodied AI technology. Our advanced AI software and foundation models help vehicles perceive, understand, and navigate complex environments, improving the usability and safety of automated driving systems. Our vision is to create autonomy that propels the world forward. We build intelligent, mapless, and hardware-agnostic AI products for automakers, accelerating the transition from assisted to automated driving. In our fast-paced environment, we embrace uncertainty and complex challenges to unlock groundbreaking solutions. We value diversity, welcome new perspectives, and foster an inclusive workplace where we back each other to deliver impact. You will join our AI Performance Tooling team within our AI Performance organization, working across cloud and embedded hardware to make model training and inference faster, more efficient, and more predictable.

last updated 36 week of 2026