Machine Learning Engineer, Compiler

Overview

In this role, you own the ML compilation pipeline from model checkpoint to deployable bundle for embedded targets. You will design compiler passes and build scalable infrastructure that spans NVIDIA TensorRT and Qualcomm QNN, ensuring accuracy and latency targets are met. You collaborate with model and training teams to validate changes and push the compiler stack across architectures. Your work directly influences what runs on Wayve’s embedded driving hardware, enabling efficient, reproducible deployment. This is a hands-on position with strategic impact on the compiler ecosystem and product readiness.

Responsibilities

• Own the end-to-end ML compilation pipeline for NVIDIA TensorRT and Qualcomm QNN targets
• Design and implement compiler passes with accuracy and latency gates
• Build scalable compilation infrastructure across platforms and SoCs
• Partner with model and training teams on compilability; build regression and benchmarking
• Set technical direction and mentor on compiler design

Key requirements

• Built or owned significant parts of an ML compilation or graph-lowering pipeline
• Deep experience with quantisation in compilation (precision typing, PTQ integration, debugging accuracy loss)
• Strong Python; comfortable developing and testing compiler infrastructure in production codebases
• Proficiency in MLIR, ONNX, TensorRT, Qualcomm QNN, or PyTorch graph capture/export
• Experience with multi-target compilation or graph partitioning across hardware backends
• Ability to reason about correctness and performance trade-offs at each compiler stage

• Clear communicator to align cross-functional teams
• Mentoring on compiler design
• Strategic thinking with ability to set direction
• MLIR
• ONNX
• TensorRT

Full job description from WhatJobs. Apply for this role →