Tile-Based AI Compiler Architect (Equity Eligible)

NVIDIA; Oregon (WI); On-site; USD 152,000 - 242,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is hiring for a software engineer on the CUDA Tile team. You will design and implement compiler transformations, develop MLIR-based dialects, and optimize tile-based kernels to run efficiently across multiple NVIDIA GPU generations.

Responsibilities include defining public APIs, crafting compiler and optimization techniques, and performing performance analysis within a dynamic, product-focused team. Strong C/C++ skills and collaboration are essential.

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

Responsibilities

• Design and implement compiler transformations for CUDA Tile.
• Develop MLIR-based dialects and lowering passes.
• Optimize performance of tile-based kernels across NVIDIA GPU generations.
• Define public APIs and contribute to compiler and optimization techniques.

Skills

C/C++ programming

Compiler optimization

IR design

Performance analysis

Test design

Education

Bachelor's/Master's/PhD in CS/CE or related field

Tools

MLIR

LLVM

XLA

TVM

Job description

NVIDIA is hiring for a software engineer on the CUDA Tile team. You will design and implement compiler transformations, develop MLIR-based dialects, and optimize tile-based kernels to run efficiently across multiple NVIDIA GPU generations.

Responsibilities include defining public APIs, crafting compiler and optimization techniques, and performing performance analysis within a dynamic, product-focused team. Strong C/C++ skills and collaboration are essential.