Machine Learning Engineer, Performance Tooling

Wayve is building autonomous driving technology that runs on real vehicles. Getting our models onto embedded hardware — correctly, quickly, and reproducibly — is one of the hardest problems between research and product.

As a ML Compiler Engineer, you will own the compilation pipeline that makes that possible. You will build and extend Wayve's ML compiler end-to-end: designing passes, integrating with vendor toolchains like NVIDIA TensorRT and Qualcomm QNN, and delivering deployable bundles that meet our accuracy and latency requirements on every target platform.

Each stage in the pipeline — capture, decomposition, precision assignment, legalisation, partitioning — can affect accuracy, latency, or whether a vendor backend accepts the graph. Your work spans the full lowering stack, building compiler passes and infrastructure that scale across architectures and target platforms.

You have built or significantly extended ML compilation or graph-lowering pipelines.

You understand multi-stage lowering (capture, decomposition, precision assignment, legalisation) and can debug what breaks at each stage.

Strong proficiency with at least one relevant stack (e.g. MLIR, ONNX, TensorRT, Qualcomm QNN, PyTorch export/capture) and confidence learning adjacent frameworks quickly.

Experience with quantisation in compilation — precision typing, PTQ integration, and tracking down accuracy loss from compiler transforms.

Comfortable from high-level model graphs down to vendor backend constraints; strong Python, with C++ a plus.

Clear communicator who can align cross-functional teams on compilation trade-offs.

Real compiler ownership — full lowering pipeline from checkpoint to deployable bundle, working deeply with TensorRT and QNN.

Hard problems — quantisation preservation through decomposition, cross-SoC precision typing, graph partitioning under speed/accuracy trade-offs, legalisation that does not silently break earlier passes.

Vehicle impact — compiler passes determine what runs on embedded hardware in Wayve's driving product.

Greenfield at Staff level — small team, high leverage, shaping the compilation stack from early stages.

Scalable infrastructure — building pipelines that work across platforms and architectures without starting from scratch each time.

Own the ML compilation pipeline end-to-end on NVIDIA (TensorRT) and Qualcomm (QNN) targets.

Design and implement compiler passes with accuracy and latency gates.

Extend precision typing and graph-splitting logic for new architectures and SoCs.

Partner with model and training teams on compilability.

Build regression and benchmarking to validate changes across releases.

Set technical direction and mentor on compiler design.