AI Computing Software Intern, GPU Kernel Libraries - 2027
NVIDIA· Intern
Shanghai, China, Beijing, ChinainternPosted 4w ago
About the role
Compute/DL Architecture Performance Optimization Intern role focused on exploring computer architectures for deep learning, optimizing GPU kernel performance, and delivering scalable modular infrastructure for NVIDIA software libraries used in training and inference.
Responsibilities
- Develop high performance operators on NVIDIA GPUs for cuBLAS, TensorRT, cuDNN, cuSparse and cuTensor libraries
- Analyze the performance of various GPU kernels on existing/new architecture, identify bottlenecks and propose creative solutions to improve them
- Design and develop software for kernel authoring and shipping
- Adopt cutting-edge AI technologies in GPU kernel or similar development workflow
Qualifications
- Pursuing a B.S., M.S., or PhD degree in computer science (or similar)
- Strong programming skills in C/C++ and Python development
- Familiar with GPU programming model and CUDA
- Good understanding about compiler technologies and experience with LLVM and MLIR
- Excellent problem solving skills, good communication and teamwork
Skills
Required:C/C++ programmingPythonCUDA
Nice to have:LLVMMLIR
Last updated Sep 25, 2026. Interested? Apply on the company site.
About NVIDIA
World Leader in AI Computing
NVIDIA invents the GPU and drives advances in AI, HPC, gaming, creative design, autonomous vehicles, and robotics.
AIautonomous vehiclesgaminggpugraphics cardhigh performance computingliquid-ai-recruiting-targetllm
More open roles at NVIDIA
- Senior Physical Verification Engineer3d agoEngineer, ASIC
- Senior System Software Engineer, Display Security3d agoEngineer, Sys SW
- Senior FPGA Prototyping Engineer - Hardware3d agoSanta Clara, CA · Engineer, ASIC
- Senior Product Engineer3d agoHybrid (Thailand) · Engineer, Product Development
- Senior Architect - Data Processing Systems, Analytics and Data Intelligence3d agoHybrid (Remote / Local) · Architect