Machine Learning Engineer (Performance Tooling)

• You’ll join the AI Performance Tooling team within Wayve’s AI Performance organization, which makes model training and inference faster, more efficient, and more predictable across cloud and embedded hardware.
• Our mission is to enable data-driven AI performance decisions across priority workloads and hardware targets: Measure performance teams can trust; Monitor trends and catch regressions; Predict the cost of changes before we run them; Advise on bottlenecks and prioritized opportunities.
• You’ll build tools that reason across the AI stack — models and operators, compilers, runtimes, accelerators, and distributed training infrastructure — turning 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.
• You’ll work closely with model, compiler, runtime, platform, and hardware teams, bringing a cross-stack view that turns measurement into clear recommendations
• 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

Benefits

• Private healthcare: Choose our optional health insurance for comprehensive coverage for you and your family.
• Paid time off: Paid vacation plus public holidays and additional leave programs, ensuring you have time to unwind.
• Mental health resources: Through Spill, you can access therapy and mental health support.
• Community and socials: Join clubs or attend team socials to connect over hobbies, sports, or just for fun.
• Competitive compensation: Our compensation package includes cash and equity, making you a true partner in our success.
• Learning and development: Budgets for books, courses, and company-wide training to support your continuous growth.- Hands-on experience developing deep learning models with PyTorch
• Judgment to turn an ambiguous performance question into a measurable one, and to prioritize what matters
• Deep, hands-on performance engineering in complex systems: profiling, roofline analysis, latency and throughput optimization, and root-causing what limits a workload
• Quantitative communication clear enough to influence another team’s priorities
• A track record of owning 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
• Data analysis skills to turn noisy measurements into conclusions you can defend