Senior AI Compiler Engineer — Drive Deep Learning Inference

NVIDIA; Town of Texas (WI); On-site; USD 152,000 - 241,500; Full time

Benefits offered by this job

Stock options

Benefits

Paid time off

Job summary

NVIDIA in the United States is seeking an AI & Deep Learning Compiler Engineer to advance the Deep Learning & AI Compiler team. You will analyze networks, optimize compiler performance, and implement AI-focused techniques for NVIDIA GPUs across data centers, devices, and robotics.

The role requires strong CUDA programming, C/C++, Python, and experience with MLIR/LLVM/Triton. You will work independently, define project goals, and contribute to public APIs and future NVIDIA AI workloads.

Qualifications

• Bachelor’s, master’s or Ph.D. in Computer Science, Computer Engineering, related field or equivalent experience.
• 3+ years of relevant work or research experience in performance analysis and compiler optimizations.
• Experience with compiler technologies (e.g., MLIR, LLVM, XLA, Triton, etc.).
• Excellent C/C++ and Python programming and software design skills, including debugging, performance analysis, and test design.
• Ability to work independently, define project goals and scope, and lead your own development efforts.
• Strong interpersonal skills are required along with the ability to work in a dynamic product-oriented team.

Responsibilities

• Analyzing deep learning networks and developing compiler optimization algorithms.
• Strong programming skills in CUDA including analyzing and debugging performance bottlenecks on GPUs.
• Scope of these efforts includes defining public APIs, performance optimizations and analysis, crafting and implementing compiler techniques for AI workloads and future NVIDIA GPUs.

Skills

CUDA programming

C/C++

Python

Education

Bachelor/Master/PhD in Computer Science or related field

Tools

MLIR

LLVM

XLA

Triton

Job description

NVIDIA in the United States is seeking an AI & Deep Learning Compiler Engineer to advance the Deep Learning & AI Compiler team. You will analyze networks, optimize compiler performance, and implement AI-focused techniques for NVIDIA GPUs across data centers, devices, and robotics.

The role requires strong CUDA programming, C/C++, Python, and experience with MLIR/LLVM/Triton. You will work independently, define project goals, and contribute to public APIs and future NVIDIA AI workloads.