ML Performance Tooling Engineer — Optimize AI Latency

Wayve; London (KY); Hybrid; USD 120,000 - 180,000; Full time

Job summary

Wayve is seeking a Software Engineer, AI Performance Tooling to join the AI Performance Tooling team, accelerating model training and inference across cloud and embedded hardware. You will design self-service performance tools, predict latency and compute costs, and work with model, compiler, runtime, platform, and hardware teams to turn measurements into actionable recommendations.

This full-time, hybrid in-office role embeds you in a fast-moving environment where ownership and cross-team

Qualifications

• Hands-on performance engineering in complex systems.
• Own a tool or service end to end — design, delivery, and adoption by other teams.
• Strong Python skills, and comfort profiling and instrumenting large production codebases.
• Hands-on experience developing deep learning models with PyTorch.
• Data analysis skills to turn noisy measurements into conclusions you can defend.
• Judgment to turn an ambiguous performance question into a measurable one, and to prioritize what matters.
• Quantitative communication clear enough to influence another team’s priorities.

Responsibilities

• Design and build reliable, self-service performance tools that scale across models, hardware targets, and development workflows.
• Shape how Wayve measures and predicts AI performance, and set the standards other teams build on.
• Model theoretical peak for a platform, compare with achieved performance, and pinpoint where efficiency is lost at layer and op level.
• Predict latency, memory, utilization, and compute cost of a model or recipe change before spending compute.
• 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.

Skills

Performance engineering

tool ownership

Python profiling

PyTorch

Data analysis

Prioritization

Cross-team collaboration

Tools

Python

PyTorch

Job description

Wayve is seeking a Software Engineer, AI Performance Tooling to join the AI Performance Tooling team, accelerating model training and inference across cloud and embedded hardware. You will design self-service performance tools, predict latency and compute costs, and work with model, compiler, runtime, platform, and hardware teams to turn measurements into actionable recommendations.

This full-time, hybrid in-office role embeds you in a fast-moving environment where ownership and cross-team