Machine Learning Jobs

• Cambridge
• Hybrid
• Senior

Develops algorithms and optimizations for NVIDIA's LPX inference and compiler stack, building high-performance runtime and compiler components and mapping large-scale neural network workloads onto future hardware. Involves LLVM/MLIR work, graph transformations, scheduling and memory/layout optimizations, plus close collaboration with hardware architects on co-design.

LLVM · MLIR · TensorFlow +3

• United Kingdom
• Hybrid
• Senior

Develop algorithms and optimizations for NVIDIA's LPX inference and compiler stack, working at the intersection of large-scale systems, compilers, and deep learning. The role involves building high-performance runtime and compiler components, mapping inference workloads onto future hardware, and collaborating with hardware architects on codesign. Distinct focus on LLVM/MLIR, graph transformations, and spatial accelerators.

LLVM · MLIR · TensorFlow +3