DL Compiler Engineer, CUDA Tile & LPU — Optimize MLIR

NVIDIA Corporation; Washington; On-site; USD 152,000 - 242,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is seeking software engineers for the CUDA Tile team in the United States. You will design and implement compiler transformations, develop MLIR-based dialects and lowering passes, and optimize tile-based kernels for broad hardware compatibility.

We expect 3+ years in compiler optimization and strong C/C++ skills, with the ability to lead your own development efforts and work in a fast-paced, product-focused team. Equity and benefits are included in the compensation package.

Qualifications

• Bachelor's, Master's or Ph.D. in CS/CE or 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.

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 and LPU architectures.

Skills

Compiler optimization

Performance analysis

IR design

C/C++

Independent leadership

Education

Bachelor/Master/PhD in CS/CE or related

Tools

MLIR

LLVM

XLA

TVM

Job description

NVIDIA is seeking software engineers for the CUDA Tile team in the United States. You will design and implement compiler transformations, develop MLIR-based dialects and lowering passes, and optimize tile-based kernels for broad hardware compatibility.

We expect 3+ years in compiler optimization and strong C/C++ skills, with the ability to lead your own development efforts and work in a fast-paced, product-focused team. Equity and benefits are included in the compensation package.