GPU Compiler Intern — NVIDIA (ufficio Zurich)
- Location
- Zürich
- Contract
- Full-time
- Posted
Full description
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.
Questions about this listing
What salary does NVIDIA (ufficio Zurich) offer for this role?
The original listing does not state a salary. Our estimate for this role and sector in Zürich is CHF 57'500 - 87'000 gross per year — an estimate, not a figure communicated by NVIDIA (ufficio Zurich). The calculator on this site converts a gross salary into your actual net take-home once cross-border tax and social contributions are applied.
Is this a full-time role, and what type of contract does NVIDIA (ufficio Zurich) offer?
This listing is a full-time position. The contract type shown here comes directly from the employer's original posting; always confirm exact hours, notice period and probation length with NVIDIA (ufficio Zurich) during the application process, since these details can vary by role even within the same contract category.
Do I need a cross-border work permit for a role in Zürich?
EU/EFTA citizens living in an EU/EFTA state can apply for a G permit and can work anywhere in Switzerland: border zones no longer apply to them, and they must return home at least once a week. The Swiss employer files the request with Canton Zürich's migration office after the contract is signed. Processing times vary by canton, so confirm the specifics with Zürich's cantonal migration office or with HR during the application.
How do I apply for this position at NVIDIA (ufficio Zurich)?
Use the "Apply now" button on this page — it links directly to NVIDIA (ufficio Zurich)'s original listing at nvidia.wd5.myworkdayjobs.com, so your application goes straight to the employer's own applicant-tracking system. Frontaliere Ticino does not collect or forward applications itself.
