Senior DL Compiler Engineer, MLIR Tile & CUDA Equity

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

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA in California is seeking software engineers for the CUDA Tile team to design and optimize a tile-based programming model across multiple GPU generations. You will design and implement compiler transformations, develop MLIR-based dialects, and improve performance of tile-based kernels, collaborating with a world-class team on AI and deep learning platforms.

You should have strong C/C++ and a proven track record in compiler work, with the ability to own projects end-to-end and contribute

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 the performance of tile-based kernels across multiple GPU generations.

Skills

C/C++ programming

Compiler optimization

Performance analysis

IR design

Independent work

Education

Bachelor's/Master's/PhD in CS/CE

Tools

MLIR

LLVM

XLA

TVM

CUDA

OpenCL

Job description

NVIDIA in California is seeking software engineers for the CUDA Tile team to design and optimize a tile-based programming model across multiple GPU generations. You will design and implement compiler transformations, develop MLIR-based dialects, and improve performance of tile-based kernels, collaborating with a world-class team on AI and deep learning platforms.

You should have strong C/C++ and a proven track record in compiler work, with the ability to own projects end-to-end and contribute