Deep Learning Compiler Engineer - CUDA
NVIDIAJob Title
Deep Learning Compiler Engineer - CUDA
Role Summary
Work on the design and implementation of a tile-aware GPU programming model: its DSL and core compiler, with a focus on performance for emerging GPU architectures. The role sits in NVIDIA's Architecture group, which develops parallel programming models, compiler infrastructure, and architecture features to maximize GPU performance.
The engineer will analyze AI/LLM workloads, iterate on compiler architecture, and propose compiler- and DSL-level solutions for next-generation GPUs.
Experience Level
Entry-level. The posting requests 2+ years of relevant work experience.
Responsibilities
Primary engineering and architecture tasks for the compiler and programming model:
- Design and implement the DSL and core compiler for a tile-aware GPU programming model.
- Iterate on compiler architecture to improve performance and code generation quality.
- Investigate next-generation GPU architectures and adapt the DSL/compiler stack accordingly.
- Perform performance analysis on emerging AI/LLM workloads and integrate results with AI/ML frameworks.
Requirements
Core technical requirements and preferred skills.
- Must-have: 2+ years of relevant work experience.
- Must-have: Strong C/C++ programming and general software engineering skills.
- Must-have: Solid fundamentals in computer architecture and problem abstraction.
- Preferred: Compiler experience (knowledge of MLIR/TVM/Triton/LLVM is desired).
- Preferred: Knowledge of GPU architecture and fast kernel programming.
- Preferred: Familiarity with LLM algorithms or HPC domains; multi-GPU distributed communication experience is a plus.
- Preferred: Good oral communication in English.
Education Requirements
Master's or PhD in Computer Engineering, Computer Science & Engineering, Computer Science, AI, or equivalent practical experience. Equivalent practical experience in related technical fields is acceptable.
About the Company
Company: NVIDIA
Headquarters: Santa Clara, California, USA
NVIDIA is a global leader in accelerated computing, renowned for its innovative solutions in AI and digital twins that transform diverse industries. The company specializes in networking technologies, providing end-to-end InfiniBand and Ethernet solutions for servers and storage that optimize performance and scalability. NVIDIA serves sectors such as high-performance computing, enterprise data centers, and cloud computing, constantly reinventing its products and services to stay ahead in the market.
