← All roles

Software Engineering Intern, DLFW Comms - 2027

NVIDIA· Intern
Apply on company site
Shanghai, Beijing, ShenzheninternPosted 4w ago

About the role

Motivated deep learning engineer internship focused on integrating advanced communication technologies into AI stacks and optimizing multi-GPU performance for AI workloads, with exposure to frameworks like PyTorch, vLLM, SGLang, TRT-LLM, and veRL.

Responsibilities

  • Integrate new communication libraries features in AI frameworks: from PoC to performance analysis to production.
  • Perform deep analysis of AI workloads and frameworks to identify multi-GPU communication requirements and opportunities. Collaborate hands-on with teams working on the latest AI models.
  • Author custom communication or fused compute-communication kernels to showcase ultimate performance on NV platforms.
  • Conduct in-depth research to achieve SOL GPU performance.
  • Build fault-tolerant and elastic solutions for large-scale or dynamic AI workloads.
  • Collaborate with a very dynamic team across multiple time zones.

Qualifications

  • You are pursuing a M.S. or Ph.D. in CE/CS/EE with a strong background in communication, kernel authoring, and/or AI training/inference.
  • Rapid prototyping and development with Python, C++, CUDA or related DSLs (Triton, cuTe).
  • Solid understanding of LLM models and parallelisms.
  • Adaptability and passion to learn new areas and tools.
  • Flexibility to work and communicate effectively.

Skills

Required:PythonC++CUDALLM models and parallelisms
Nice to have:Python tooling for ML frameworks

Benefits

  • Widely considered to be one of the technology world’s most desirable employers, NVIDIA offers highly competitive salaries and a comprehensive benefits package.
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
Company profileWebsite

More open roles at NVIDIA