AI Computing Software Intern, GPU Kernel Libraries - 2027
Core
Develop high-performance GPU operators and scalable infrastructure for NVIDIA's deep learning libraries (cuBLAS, TensorRT, cuDNN, cuSparse, cuTensor) to optimize training and inference.
Role type
Intern, GPU kernel library developer
Builds
Optimized GPU operators and modular infrastructure for NVIDIA software libraries
Domain
AI computing, GPU architecture, deep learning
Deliverable
production ML models
Required skills
C/C++, Python, CUDA, GPU programming model, compiler technologies, LLVM, MLIR
Preferred skills
Deep learning operator optimization, fusion operator design
Technologies
cuBLAS, TensorRT, cuDNN, cuSparse, cuTensor, CUDA, LLVM, MLIR
Responsibilities
Develop high-performance operators on NVIDIA GPUs; Analyze GPU kernel performance and identify bottlenecks; Design and develop software for kernel authoring and shipping; Adopt cutting-edge AI technologies in GPU kernel development workflow
Seniority
Intern
