Startups · AI

Machine Learning Engineer, Performance Tooling

Wayve · London; Sunnyvale · On-site

← All jobs
About Wayve

Backed by SoftBank VF and Air Street.

About the role

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.

What they're looking for

  • 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
More about this role

Founded in 2017, Wayve is the leading developer of Embodied AI technology. Our advanced AI software and foundation models enable vehicles to perceive, understand, and navigate any complex environment, enhancing the usability and safety of automated driving systems.

Our vision is to create autonomy that propels the world forward. Our intelligent, mapless, and hardware-agnostic AI products are designed for automakers, accelerating the transition from assisted to automated driving.

In our fast-paced environment big problems ignite us—we embrace uncertainty, leaning into complex challenges to unlock groundbreaking solutions. We aim high and stay humble in our pursuit of excellence, constantly learning and evolving as we pave the way for a smarter, safer future.

At Wayve, your contributions matter. We value diversity, embrace new perspectives, and foster an inclusive work environment; we back each other to deliver impact.

Make Wayve the experience that defines your career!

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...

Read the full posting on Wayve's site ↗

AI Platform

Build your edge while you search

Free tools for founders and investors, plus VC Unfiltered, our take on startups, venture and the people who build them.