CareerPlan

AI Computing Software Intern, GPU Kernel Libraries - 2027

China, ShanghaiInternship2026-09-23 → 2026-11-10

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