AI & Deep Learning Compiler Architect

NVIDIA; Washington (Washington County); On-site; USD 152,000 - 241,500; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is seeking an AI & Deep Learning Compiler Engineer to join the DLC team. You will analyze networks, optimize compiler performance, and implement techniques for AI workloads on NVIDIA GPUs. Strong CUDA, C/C++, Python skills and experience with MLIR/LLVM/XLA/Triton are essential.

Ideal candidates will lead development efforts, work independently, and collaborate in a fast-paced product-driven environment. Equity and comprehensive benefits accompany a competitive base salary.

Qualifications

• Bachelor's degree in CS/CE or related field; equivalent experience accepted.
• 3+ years in performance analysis and compiler optimizations.
• Experience with compiler technologies (MLIR, LLVM, XLA, Triton, etc.).
• Excellent C/C++ and Python programming and software design; debugging and performance analysis skills.

Responsibilities

• Analyze deep learning networks and develop compiler optimization algorithms.
• Debug and optimize CUDA performance bottlenecks on GPUs.
• Define public APIs and implement compiler techniques for AI workloads and future NVIDIA GPUs.

Skills

CUDA

C/C++

Python

Performance analysis

Debugging

MLIR

LLVM

XLA

Triton

Education

Bachelor's degree in Computer Science/Engineering or related field

Master's or Ph.D. preferred

Equivalent experience

Tools

MLIR

LLVM

XLA

Triton

Job description

NVIDIA is seeking an AI & Deep Learning Compiler Engineer to join the DLC team. You will analyze networks, optimize compiler performance, and implement techniques for AI workloads on NVIDIA GPUs. Strong CUDA, C/C++, Python skills and experience with MLIR/LLVM/XLA/Triton are essential.

Ideal candidates will lead development efforts, work independently, and collaborate in a fast-paced product-driven environment. Equity and comprehensive benefits accompany a competitive base salary.