Deep Learning Compiler Engineer

NVIDIA•Redmond, WA

8d•$152,000 - $241,500

About The Position

NVIDIA is hiring software engineers for its Tensor IR & CUDA Tile team. The team works on compiler technologies for NVIDIA GPUs, which power advances in generative AI, large language models, recommendation systems, speech recognition, image classification, and other fields. In this role, you will work on CUDA Tile, a tile-based programming model shipped with CUDA 13.1, and TensorIR, an open-source compiler infrastructure project that uses CUDA Tile to generate high-performance GPU kernels. You will design and implement compiler transformations, develop MLIR-based dialects and lowering passes, and optimize tile-based kernels so they execute efficiently across multiple generations of NVIDIA GPU architectures. The work also includes defining public APIs, developing compiler and optimization techniques, performance optimization, and general software engineering.

Requirements

Bachelor’s, Master’s, or Ph.D. in Computer Science, Computer Engineering, or a related field, or equivalent experience.
At least 3 years of relevant work or research experience in compiler optimization, performance analysis, and IR design.
Ability to work independently, define project goals and scope, and lead your own development effort.
Excellent C/C++ programming and software design skills, including debugging, performance analysis, and test design.
Strong interpersonal skills and ability to work in a dynamic, product-oriented team.

Nice To Haves

Knowledge of CPU and/or GPU architecture.
CUDA or OpenCL programming experience.
Experience with MLIR, LLVM, XLA, or TVM.
Experience with deep learning models and algorithms.

Responsibilities

Design and implement compiler transformations.
Develop MLIR-based dialects and lowering passes.
Optimize tile-based kernels for efficient execution across multiple generations of NVIDIA GPU architectures.
Define public APIs.
Craft and implement compiler and optimization techniques.
Perform performance optimization and other general software engineering work.