# Intern - AI Computing Platforms

**Company**: NVIDIA
**Experience**: internship
**Job type**: internship
**Category**: Engineering
**Industry**: Technology

**Apply**: https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/China-Shanghai/AI-Computing-Software-Development-Intern---2027_JR2026164?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply
**Canonical**: https://yubhub.co/jobs/job_c8a22826-8f9

## Description

NVIDIA is seeking interns to contribute to building AI computing platforms that drive innovation across industries. As an intern, you'll focus on one of three specialized tracks:

**TensorRT LLM – Inference Optimization (Python / PyTorch)** Build and enhance high-performance LLM inference pipelines. Analyze and optimize model execution, scalability, and memory usage. Collaborate across framework and research teams to deliver efficient multi-GPU model serving.

**TensorRT Compiler – Graph Optimization (C++)** Work on the TensorRT compiler backend to improve graph transformations and code generation for NVIDIA GPUs. Develop compiler optimization passes, refine operator fusion and memory allocation, and collaborate with CUDA and hardware architecture teams.

**CuTe DSL & CUDA Kernels Development / Optimization (CUDA C++ / Python DSL)** Design and tune GPU compute kernels and DSL implementations for core deep learning operations such as GEMM, MoE, Attention and Convolution. Profile, analyze and improve CUDA kernel performance to achieve maximum GPU efficiency.

**Requirements:**

- Pursuing an M.S. or Ph.D. in Computer Science, Computer Engineering, Electrical Engineering, Applied Mathematics, or related field.

- Excellent problem-solving ability, curiosity for cutting-edge AI systems, and passion for GPU computing and deep learning software performance.

- Specific skills required for each track:

- TensorRT LLM: Strong Python programming and experience with PyTorch; solid understanding of inference and GPU acceleration.

- TensorRT Compiler: Proficient in C++, with experience in compiler or performance optimization.

- CuTe DSL & CUDA Kernels: Skilled in C/C++ and CUDA or parallel programming; familiar with LLVM, MLIR and compiler; understanding of computer architecture and performance profiling/analysis/optimization.

## Skills

### Required
- Python
- PyTorch
- C++
- CUDA
- parallel programming
- compiler
- performance optimization
- GPU computing
- deep learning

---

Source: [Apply at nvidia.wd5.myworkdayjobs.com](https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/China-Shanghai/AI-Computing-Software-Development-Intern---2027_JR2026164?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply)