Senior AI Compiler Engineer, Algorithms and Code-Generation

NVIDIA is seeking a Senior AI Compiler Engineer to join its Deep Learning & AI Compiler (DLC) team. The ideal candidate will analyze deep learning networks, develop compiler optimization algorithms, and work on performance optimizations and analysis.

Responsibilities:

• Analyze deep learning networks and develop compiler optimization algorithms.
• Develop strong programming skills in CUDA, including analyzing and debugging performance bottlenecks on GPUs.
• Define public APIs, performance optimizations and analysis, craft and implement compiler techniques for AI workloads and future NVIDIA GPUs.

Requirements:

• 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 development efforts.
• Strong interpersonal skills are required along with the ability to work in a dynamic product-oriented team.

Preferred Qualifications:

• Proficient in CPU and/or GPU architecture, especially modern Nvidia GPUs like Hopper and Blackwell.
• Understanding of deep learning models, algorithms, and frameworks, such as PyTorch, JAX.
• GPU kernel authoring and performance analysis using tools such as Nsight Compute.
• A track record of success in mentoring early-career engineers and interns.
• Track record on new hardware bring-up.

Benefits:

• Highly competitive salaries
• Comprehensive benefits package
• Equity eligibility