Senior Compiler Engineer, AI Inference Platforms
NVIDIA · Austin, TX · 1 wk ago
EngineeringFull-time
About the role
NVIDIA is seeking an AI & Deep Learning Compiler Engineer to join the Deep Learning & AI Compiler (DLC) team. The DLC is crucial for delivering leading inference performance, fast build times, reduced memory footprints, and ease of use through Ahead-of-Time and Just-in-Time compilation.
Responsibilities
- Analyzing deep learning networks and developing compiler optimization algorithms.
- Collaborating with members of the deep learning software framework teams and the GPU architecture teams to accelerate the next generation of deep learning software.
- Defining public APIs, performance optimizations, and analysis.
- Crafting and implementing 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 your own development efforts.
- Strong interpersonal skills are required along with the ability to work in a dynamic product-oriented team.
Qualifications
- Proficient in CPU and/or GPU architecture.
- CUDA or OpenCL programming experience.
- Understanding of deep learning models, algorithms and frameworks, such as PyTorch, JAX.
- GPU kernel authoring and performance analysis using tools such as Nsight Compute.
Skills
- Track record of success in mentoring early-career engineers and interns is a bonus.
- Track record on new hardware bring-up is a plus.
Benefits
Competitive salaries and a generous benefits package are offered. Applications for this job will be accepted at least until February 27, 2026. NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer.