Senior Software Engineer, Dl Compilers

NVIDIA NVIDIA · Semiconductors · Santa Clara, CA +5 · Remote

Senior Software Engineer role focused on building the code generation backend for NVIDIA's deep learning compilers, connecting ML frontends to GPU compilation for high-performance kernel generation.

What you'd actually do

  1. Designing and implementing compiler passes, IRs, and lowering pipelines for GPU kernel generation for DL compiler and framework integration.
  2. Building MLIR-based transformations and compiler infrastructure connecting frontend representations to backend code generation.
  3. Working within and contributing to the backend compilation stack itself, including tile-based IRs and their associated optimization and lowering infrastructure.
  4. Performance analysis and optimization across the codegen pipeline, targeting both compute-bound and memory-bound GPU kernels against competitive baselines.
  5. Making architectural decisions about how the codegen backend integrates with compiler frontends, GPU libraries, and NVIDIA's broader compilation ecosystem.

Skills

Required

  • BS, MS, or PhD in Computer Science, Computer Engineering, or a related field (or equivalent experience).
  • 6+ years of relevant work or research experience in compilers, with focus on code generation, IR design, or optimization passes.
  • Strong C/C++ skills, including debugging, performance profiling, and designing for maintainability.
  • Ability to work independently and drive projects with increasing scope and ambiguity.
  • Strong interpersonal and communication skills, including the ability to work across teams and with external partners.

Nice to have

  • Hands-on MLIR experience: designing dialects, writing passes, and reasoning about abstraction boundaries in a compilation pipeline.
  • Track record of owning and delivering complex compiler infrastructure end to end.
  • Working experience with GPU or other high-performance accelerator architectures and execution models.
  • Contributions to open-source compiler projects (MLIR, LLVM, XLA, TVM, OAI Triton).
  • History of mentoring engineers and raising the technical bar on a team.

What the JD emphasized

  • 6+ years of relevant work or research experience in compilers, with focus on code generation, IR design, or optimization passes.
  • Strong C/C++ skills, including debugging, performance profiling, and designing for maintainability.
  • Track record of owning and delivering complex compiler infrastructure end to end.

Other signals

  • compiler engineering
  • GPU kernel generation
  • deep learning workloads