Senior Software Engineer, DL Compilers

Job description

NVIDIA's GPUs are at the core of modern AI infrastructure, from training large-scale models to running inference in production. That position depends on software as much as hardware, and compiler engineering is a big part of what makes it work. We're hiring senior software engineers for a compiler team within NVIDIA's deep learning software organization. This team builds a code generation backend that connects frontend ML compilers to NVIDIA's GPU compilation and execution stack, targeting high-performance kernel generation for deep learning workloads. What you'll be doing: You'll work on the compiler infrastructure responsible for generating high-performance GPU kernels from frontend compiler representations. This spans the full codegen pipeline: taking in high-level graph operations, lowering them through intermediate representations, and producing efficient code targeting NVIDIA's GPU compiler backends. You may work at any layer of this stack, from the interface with compiler frontends down to the tile-level IR and code generation machinery underneath. Designing and implementing compiler passes, IRs, and lowering pipelines for GPU kernel generation for DL compiler and framework...