Deep Learning Compiler Engineer

About the role

We are hiring software engineers for the CUDA Tile team. NVIDIA GPUs are at the center of the deep learning revolution and continue to enable breakthroughs in generative AI, large language models, recommendation systems, speech recognition, image classification and other areas. Come join us to work with a top-notch team and have broad impact across the entire deep learning community.

In this role, you will be working on CUDA Tile, a new tile-based programming model for our GPUs. CUDA Tile shipped with CUDA 13.1 and is a major addition to CUDA (developer.nvidia.com/cuda/tile). You will design and implement compiler transformations, develop MLIR-based dialects and lowering passes, and optimize the performance of tile-based kernels to ensure they execute efficiently across multiple generations of NVIDIA GPU architectures. The scope of these efforts includes defining public APIs, crafting and implementing compiler and optimization techniques, performance optimization, and other general software engineering work.

Responsibilities

• Design and implement compiler transformations for CUDA Tile.
• Develop MLIR-based dialects and lowering passes.
• Optimize the performance of tile-based kernels across multiple generations of NVIDIA GPU architectures.
• Define public APIs and craft compiler and optimization techniques.
• Perform general software engineering tasks, including debugging, performance analysis, and test design.

Requirements

• Bachelor’s, Master’s, or Ph.D. in Computer Science, Computer Engineering, or a related field (or equivalent experience).
• 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 the ability to work in a dynamic product-oriented team.

Skills

Preferred:

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

Benefits

NVIDIA offers a comprehensive benefits package and is widely considered to be one of the technology industry's most desirable employers. Our product lines are growing fast in cutting-edge fields such as Virtual Reality, Artificial Intelligence, Deep Learning, and Autonomous Vehicles.

Pay

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is $152,000 USD - $241,500 USD. You will also be eligible for equity and benefits.