Performance Software Intern, Deep Learning Libraries - 2027
NVIDIA· Intern
Shanghai, BeijingInternPosted 4w ago
About the role
Performance software internship focusing on developing optimized kernels for deep learning operations on NVIDIA GPUs, targeting deep learning libraries and low-level software stack optimization.
Responsibilities
- Writing highly tuned compute kernels to perform core deep learning operations (e.g. matrix multiplies, MoE, Attention)
- Following general software engineering best practices including support for regression testing and CI/CD flows
- Collaborating with teams across NVIDIA: Compiler team on generating optimal assembly code; Deep learning training and inference performance teams on which layers require optimization; Hardware and architecture teams on the programming model for new deep learning hardware features
Qualifications
- Pursuing Masters or PhD degree in Computer Science, Computer Engineering, Applied Math, or related field
- Demonstrated strong programming and software design skills, including debugging, performance analysis, and test design
- Experience with performance-oriented parallel programming, even if it’s not on GPUs (e.g. with OpenMP or pthreads)
- Solid understanding of computer architecture and some experience with assembly programming
- Identify bottlenecks, optimize resource utilization, and improve throughput.
Skills
Required:Programming and software designPerformance-oriented parallel programmingComputer architecture
Nice to have:Kernel optimization / CUDA GPU programming
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