Senior Dl Compiler Engineer -cuda Tile

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

NVIDIA is seeking a Senior DL Compiler Engineer to work on CUDA Tile, a new tile-based programming model for GPUs. The role involves designing and implementing compiler transformations, developing MLIR-based dialects, and optimizing kernel performance for efficient execution on NVIDIA GPU architectures. This includes defining APIs, crafting optimization techniques, and general software engineering.

What you'd actually do

  1. design and implement compiler transformations
  2. develop MLIR-based dialects and lowering passes
  3. optimize the performance of tile-based kernels to ensure they execute efficiently across multiple generations of NVIDIA GPU architectures
  4. defining public APIs
  5. crafting and implementing compiler and optimization techniques, performance optimization, and other general software engineering work

Skills

Required

  • compiler optimization
  • performance analysis
  • IR design
  • C/C++ programming
  • software design
  • debugging
  • test design

Nice to have

  • CPU and/or GPU architecture knowledge
  • CUDA programming experience
  • OpenCL programming experience
  • MLIR
  • LLVM
  • XLA
  • TVM
  • deep learning models and algorithms

What the JD emphasized

  • 3+ years of relevant work or research experience in compiler optimization, performance analysis and IR design
  • Excellent C/C++ programming and software design skills, including debugging, performance analysis, and test design

Other signals

  • CUDA Tile shipped with CUDA 13.1
  • new tile-based programming model for our GPUs
  • compiler transformations
  • MLIR-based dialects and lowering passes
  • optimize the performance of tile-based kernels