CareerPlan

Senior Machine Learning Applications and Compiler Engineer, LPX

UK, CambridgeFull-time2026-05-13 → 2026-11-10
Hybrid

Core

Develop algorithms and optimizations for NVIDIA's LPX inference and compiler stack, mapping neural network workloads onto future NVIDIA platforms.

Role type

Senior IC machine learning applications and compiler engineer

Builds

High-performance runtime and compiler components for end-to-end inference optimization

Domain

Deep learning systems, compiler technology, and spatial processor architectures

Deliverable

production ML models

Required skills

Systems-level programming (C/C++/Rust), compiler/runtime development (IR design, optimization passes, code generation), LLVM/MLIR expertise, deep learning frameworks (TensorFlow/PyTorch/ONNX), parallel/heterogeneous compute architectures, profiling and benchmarking

Preferred skills

Spatial/dataflow architectures (static scheduling, pipeline/tensor parallelism), open-source contributions to ML frameworks/compilers, research publications at top-tier venues, large-scale AI distributed inference systems

Technologies

LLVM, MLIR, TensorFlow, PyTorch, ONNX, C, C++, Rust