← All roles

AI Infrastructure and Frameworks Intern, Cosmos Lab - 2027

NVIDIA· Intern
Apply on company site
Hybrid (Beijing), Hybrid (Shanghai), Hybrid (Shenzhen)internPosted 1mo ago

About the role

Internship in NVIDIA Cosmos Lab Infrastructure focusing on developing and optimizing training and post-training systems for advanced Physical AI models, including world foundation models and robot policies, with hands-on projects on NVIDIA GPU infrastructure.

Responsibilities

  • Develop and optimize training infrastructure for advanced Physical AI world models, supporting pre-training, supervised fine-tuning (SFT), and reinforcement learning (RL). Explore distributed parallelism, sharding, low-precision training, compute–communication overlap, and numerical consistency and efficient weight synchronization between training and inference.
  • Build Physical AI post-training and RL infrastructure supporting advanced training algorithms. Connect simulation or real-robot interaction with experience collection, rollout inference, reward computation, training, and evaluation. Optimize workflows through partitioning, pipelining, data transfer, and synchronization across synchronous, asynchronous, or disaggregated execution.
  • Improve efficiency and scalability across training, inference, simulation, and evaluation through scheduling, placement, dynamic resource allocation, and load balancing, supporting heterogeneous resources, elasticity, and fault recovery.
  • Analyze and optimize system performance, working with researchers to investigate, support, and compare emerging Physical AI models, training workflows, and algorithms from a systems perspective. Use profiling, benchmarking, and performance modeling to identify bottlenecks and measure throughput, latency, GPU utilization, and policy freshness. Share findings through tested code, documentation, and technical presentations, and contribute to research publications where appropriate.

Qualifications

  • Pursuing a Bachelor’s, Master’s, or PhD in Computer Science, Computer Engineering, Electrical Engineering, or a related field.
  • Strong Python and debugging skills, with systems fundamentals in concurrency, distributed execution, memory management, or data movement.
  • Practical experience in at least one area: training infrastructure, RL infrastructure, simulation or robotics integration, or inference infrastructure. Coursework, research, open-source projects, and internships all count.
  • Strong analytical and communication skills, curiosity, and a willingness to learn.
  • Experience in every listed area, prior access to large GPU clusters, and model architecture or learning algorithm research are not required.

Skills

Required:Pythonsystems fundamentalstraining infrastructure
Nice to have:simulation or robotics integration
Last updated Sep 18, 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