Deep Learning Compiler Engineer

NVIDIA Corporation; Santa Clara, Northern (CA, KY); Hybrid; USD 152,000 - 242,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA Corporation is hiring software engineers for the CUDA Tile team in Santa Clara, CA. You will design and implement compiler transformations, develop MLIR-based dialects and lowering passes, and optimize tile-based kernels for multiple GPU generations.

You will contribute to public APIs, compiler optimizations, and general software engineering tasks, with emphasis on performance analysis and debugging. A strong C/C++ background and teamwork are essential for success.

Qualifications

• Bachelors, Masters 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 ability to work in a dynamic product-oriented team.

Responsibilities

• Design and implement compiler transformations for CUDA Tile and optimize tile-based kernels across NVIDIA GPUs.
• Develop MLIR-based dialects and lowering passes; define public APIs and implement compiler/optimization techniques.
• Contribute to performance optimization and general software engineering work for the CUDA Tile project.

Skills

C/C++ programming

Independent work

Software design

Debugging

Performance analysis

Test design

Education

Bachelor's/Master's/Ph.D. in CS/CE or related field

Tools

MLIR

LLVM

XLA

TVM

Job description

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It's a unique legacy of innovation that’s fueled by great technology—and amazing people. Today, we're tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what's never been done before takes vision, innovation, and the world's best talent. As an NVIDIAN, you'll be immersed in a diverse, supportive environment where everyone is inspired to do their best work. Come join the team and see how you can make a lasting impact on the world. 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.

What you'll be doing:

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 ( https://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.

What we need to see:

• Bachelors, Masters 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 are required along with the ability to work in a dynamic product-oriented team.

Ways to stand out from the crowd:

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.

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 some of the hottest state of the art fields such as Virtual Reality, Artificial Intelligence, Deep Learning and Autonomous Vehicles.

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.

NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.