NVIDIA seeks a Deep Learning Compiler Engineer to join its DLC team, focusing on advancing the inference engine for applications like large language models and image classification. This role involves collaborating with software framework and hardware architecture teams to enhance public APIs, optimize performance, and develop compiler infrastructure techniques for neural networks. The ideal candidate should have expertise in CPU and GPU architectures, experience with constrained resource systems, and proficiency in technologies such as MLIR, XLA, TVM, LLVM, and deep learning frameworks like PyTorch. Additionally, the ability to generate high-performance GPU kernels and mentor junior engineers is highly valued. This position plays a crucial role in delivering leading inference performance across diverse platforms including data centers and automotive systems.
Skills
MLIR
XLA
TVM
LLVM
PyTorch
CUDA
C++
Python
GPU
CPU
Embedded_Systems
Cross_Compilation
CI/CD
What you'll do
Define public APIs for deep learning software frameworks.
Optimize performance and analyze neural network compiler infrastructure.
Implement techniques to reduce memory footprints in compilers.
Generate GPU kernels with high performance and fast build times.
Work on cross-compilation and systems with constrained resources.
Mentor junior engineers and interns in the team.
What we're looking for
Bachelor’s degree or higher in computer science or related field.
Proficiency in CPU and GPU architecture optimization.
Experience working on systems with constrained resources like embedded platforms.
Expertise in MLIR, XLA, TVM, LLVM, and deep learning frameworks such as PyTorch.
Strong background in generating high-performance GPU kernels efficiently.
Track record of mentoring junior engineers and interns (preferred).
Market check
Salary context
This
$152,000–$241,500
range sits above
77%
of similar postings on FindRole.
Peer median band
$119,800–$212,505
Median floor and ceiling across peers.
Typical midpoint (25–75%)
$142,400–$196,750
Middle half of comparable postings.
Based on 240 comparable postings.*
* 240 is the maximum number of comparable postings sampled.
Employer
About Nvidia
Nvidia is a leading designer of graphics processing units (GPUs) and system-on-chip units, powering gaming, professional visualization, data centers, and artificial intelligence workloads. Industry: Semiconductors & AI Computing
Nvidia currently has
801 open roles
on FindRole.
Listed pay typically runs
$184,000–$287,500
across 797 roles with salary data.