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).

Responsibilities

• Design and implement compiler transformations.
• Develop MLIR-based dialects and lowering passes.
• Optimize the performance of tile-based kernels to ensure they execute efficiently across multiple generations of NVIDIA GPU architectures.
• Define public APIs, craft and implement compiler and optimization techniques, and perform performance optimization.
• Engage in general software engineering work.

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 the following technologies: MLIR, LLVM, XLA, TVM, and deep learning models and algorithms.

Benefits

With highly competitive salaries and a comprehensive benefits package, NVIDIA is widely considered to be one of the technology industry’s most desirable employers. We have some of the most brilliant and hardworking people in the world working with us, and 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.