GPU Compiler Intern

NVIDIA is hiring a GPU Compiler Intern in Zurich, Switzerland. Level rates it ; you can apply on NVIDIA.

Level's read on this job

Builds AI ●●●● is how Level rates this role: the job is building AI systems.
Level classes it as an internship.
NVIDIA has 2,600+ other open roles on Level.
Posted in the last few days.

AI in this role

As GPU Compiler Intern, you will develop compiler infrastructure for Python-based GPU programming, including CUTLASS CuTeDSL. You will work on DSL design, MLIR IR design, and compiler optimizations that translate expressive Python programs into high-performance NVIDIA GPU kernels.

What you will be doing:

Design and implement Python DSL features for GPU kernel authoring, including metaprogramming and IR generation.
Develop MLIR dialects, transformation passes, and lowering pipelines that preserve the information needed for effective GPU optimization.
Design and optimize GPU kernels, exploring how IR design and compiler transformations affect generated code and performance.
Profile, benchmark, and validate compiler and kernel improvements across representative workloads.

What we need to see:

Currently pursuing or holding a Bachelor’s, Master’s, or Ph.D. degree in Computer Science, Computer Engineering, or a related technical discipline, or equivalent practical experience.
Proficiency in Python and C++, with strong debugging and software development skills.
Understanding of how Python DSLs work, including metaprogramming, staged execution, and translation into compiler IR.
Experience with MLIR, including IR construction, transformation passes, or lowering.
Practical experience with GPU programming, preferably CUDA, and knowledge of GPU execution and memory models.
Familiarity with performance analysis and optimization, including profiling, memory access patterns, and parallel execution.
Experience with CUTLASS, CuTeDSL, Triton, LLVM, or other GPU compiler and kernel development tools is a plus.

How we rate this

GPU Compiler Intern at NVIDIA rates 94 out of 100 for how much of the daily work is AI. That makes it Builds AI (AI Level 4 of 4). The level is about AI in the job, not seniority.

Classification

Builds AI. The job is building AI systems.

●●●● Builds AI80 to 100
●●●○ Works on AI60 to 79
●●○○ Uses AI40 to 59
●○○○ Little AI0 to 39

Levels come from how often the tools, models and workflows of the role are named in the posting itself. Read the description and count the AI tools, models and workflows it names.