Senior CUDA Tile Compiler Engineer - MLIR/GPU Optimizer

NVIDIA; Santa Clara (CA); On-site; USD 152,000 - 242,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is hiring software engineers for the CUDA Tile team in Santa Clara. You will design and implement compiler transformations for a tile-based model, develop MLIR-based dialects, and optimize kernel performance across generations of NVIDIA GPUs.

Qualifications include a PhD/Masters/Bachelors in CS/CE with 3+ years in compiler optimization, C/C++ programming, and strong interpersonal skills. The role emphasizes independent work and collaboration on a dynamic product team.

Qualifications

• Bachelor's, 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 tile-based kernels.
• Develop MLIR-based dialects and lowering passes.
• Optimize the performance of tile-based kernels across GPU architectures.
• Define public APIs and craft compiler/optimization techniques.
• Perform general software engineering work.

Skills

C/C++ programming

Compiler optimization

Performance analysis

IR design

Independent work

Interpersonal skills

Education

Bachelors/Masters/PhD in CS/CE or related field

Tools

MLIR

LLVM

XLA

TVM

Job description

NVIDIA is hiring software engineers for the CUDA Tile team in Santa Clara. You will design and implement compiler transformations for a tile-based model, develop MLIR-based dialects, and optimize kernel performance across generations of NVIDIA GPUs.

Qualifications include a PhD/Masters/Bachelors in CS/CE with 3+ years in compiler optimization, C/C++ programming, and strong interpersonal skills. The role emphasizes independent work and collaboration on a dynamic product team.