Senior AI Compiler Engineer: Algorithms & Code Generation

NVIDIA Corporation; Santa Clara (CA); On-site; USD 152,000 - 242,000; Full time

Job summary

NVIDIA Corporation in Santa Clara, CA is seeking an AI & Deep Learning Compiler Engineer to advance the DLC team’s compiler optimizations for DL workloads. You will analyze networks and implement techniques to improve inference speed and memory efficiency.

Requirements: 3+ years in performance analysis and compiler optimizations; strong C/C++ and Python; experience with MLIR, LLVM, XLA, Triton. You’ll influence API design and collaborate across hardware and software teams.

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.

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

C/C++

Python

Performance analysis

Compiler optimizations

Independent work

Team collaboration

Education

Bachelor's/Master's/PhD in CS/CE or related field

Tools

MLIR

LLVM

XLA

Triton

Job description

NVIDIA Corporation in Santa Clara, CA is seeking an AI & Deep Learning Compiler Engineer to advance the DLC team’s compiler optimizations for DL workloads. You will analyze networks and implement techniques to improve inference speed and memory efficiency.

Requirements: 3+ years in performance analysis and compiler optimizations; strong C/C++ and Python; experience with MLIR, LLVM, XLA, Triton. You’ll influence API design and collaborate across hardware and software teams.