Senior DL Compiler Engineer -CUDA Tile

NVIDIA; California (MO); On-site; USD 152,000 - 241,500; Full time

Benefits offered by this job

Equity and Benefits

Comprehensive salary package

Job summary

NVIDIA is seeking talented individuals for a role focused on CUDA Tile, a new programming model for GPUs. You will design and implement compiler transformations to optimize performance across generations of GPU architectures while working in a supportive team environment.

The ideal candidate has a strong background in Computer Science or related fields and has 3+ years of relevant experience in compiler optimization. NVIDIA offers competitive salaries and comprehensive benefits, promoting an inclusive workplace.

Qualifications

• 3+ years of experience in compiler optimization and performance analysis.
• Ability to define project goals and lead development efforts.
• Strong interpersonal skills to work in a dynamic team.

Responsibilities

• Design and implement compiler transformations for GPUs.
• Develop MLIR-based dialects and optimize performance of tile-based kernels.
• Define public APIs and implement optimization techniques.

Skills

C/C++ programming

Performance analysis

Compiler optimization

Education

Bachelors, Masters or Ph.D. in Computer Science or related field

Tools

MLIR

LLVM

CUDA or OpenCL

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, where 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.

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