Senior Deep Learning Compiler Engineer - HW-SW Codesign

What you’d do

Develop compiler optimization algorithms for deep learning workloads, improving inference and training performance for the JAX framework and OpenXLA compiler on NVIDIA GPUs. Includes graph partitioning, tensor sharding, performance tuning, and code generation using MLIR, LLVM, and Triton. Full time. Found on the employer's careers page on Sep 29, 2026. Apply through the link.