NVIDIA conçoit des processeurs graphiques, des systèmes de calcul et des logiciels pour l'intelligence artificielle et les applications graphiques.
GPU Compiler Intern
Vous développerez l’infrastructure de compilation pour la programmation GPU en Python, notamment avec CUTLASS CuTeDSL. Le poste couvre la conception de DSL et de dialectes MLIR, l’optimisation de kernels GPU ainsi que le profilage et la validation de leurs performances. Il s’adresse à une personne qui poursuit ou détient un diplôme en informatique ou dans une discipline technique connexe, avec des compétences en Python, C++ et programmation GPU.
Repères sur NVIDIA
- Domaine officiel
- nvidia.com
- Offres ouvertes
- 112
Détails de l’offre
La description complète publiée par NVIDIA.
Description de l’offre
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.
Prérequis
- Bachelor’s, Master’s, or Ph.D. degree in Computer Science, Computer Engineering, or a related technical discipline, or e
- Python
- C++
- Understanding of Python DSLs, metaprogramming, staged execution, and translation into compiler IR
- Experience with MLIR
- Practical experience with GPU programming, preferably CUDA
- Familiarity with performance analysis and optimization
- Experience with CUTLASS, CuTeDSL, Triton, LLVM, or other GPU compiler and kernel development tools
Candidater à cette offre
Comment candidater ?
L’offre est publiée sur le site de NVIDIA. Ouvrez l’offre originale ↗ pour y déposer votre candidature. Screeningpass ne remplit pas de formulaire à votre place.
À qui écrire chez NVIDIA pour ce poste ?
Lancez une recherche de contact depuis cette page. Screeningpass vous propose jusqu’à trois personnes liées au poste. Vous choisissez celle à qui écrire, puis Screeningpass cherche son e-mail professionnel et prépare un message. Vous le relisez et l’envoyez vous-même avec Gmail. Une adresse e-mail n’est pas toujours disponible, et une réponse n’est jamais garantie.
Combien coûte la recherche d’un contact ?
Une recherche coûte 1 crédit. Le crédit vous est rendu si la recherche échoue. L’inscription offre 3 crédits, soit 3 recherches.
À quoi sert Screeningpass ?
Screeningpass rassemble des offres de stage, d’alternance et d’emploi publiées par les entreprises. Pour une offre, Screeningpass trouve une personne à contacter dans l’entreprise, cherche son e-mail professionnel et prépare un message à partir de votre CV. Voir le fonctionnement en détail.