# NVIDIA Intern - Cosmos Lab Infrastructure

**Company**: NVIDIA
**Location**: Shanghai
**Job type**: full-time
**Category**: Engineering
**Industry**: Technology

**Apply**: https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/China-Shanghai/AI-Infrastructure-and-Frameworks-Intern--Cosmos-Lab---2027_JR2029468?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply
**Canonical**: https://yubhub.co/job/nvidia-intern-cosmos-lab-infrastructure-d47cd0ff-3e8/

## Description

Join NVIDIA's Cosmos Lab Infrastructure team to develop training and post-training systems for advanced Physical AI models. You will work on implementing and evaluating systems improvements on real AI workloads using NVIDIA's GPU infrastructure.

**Responsibilities:**

- Develop and optimize training infrastructure for advanced Physical AI world models

- Build Physical AI post-training and RL infrastructure supporting advanced training algorithms

- Improve efficiency and scalability across training, inference, simulation, and evaluation

- Analyze and optimize system performance, working with researchers to investigate emerging Physical AI models

**Requirements:**

- 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

- Strong analytical and communication skills, curiosity, and a willingness to learn

For more fresh new jobs, check out [yubhub.co](https://yubhub.co/).

## Skills

### Required
- Python
- debugging
- concurrency
- distributed execution
- memory management
- data movement

### Nice to have
- distributed parallelism
- low-precision training
- GPU memory efficiency
- compute-communication overlap
- scheduling
- placement
- resource allocation
- data transfer
- RL pipelines
- simulation environments
- robot interfaces
- inference
- GPU profiling
- C++/CUDA development

---

Source: [Apply at nvidia.wd5.myworkdayjobs.com](https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/China-Shanghai/AI-Infrastructure-and-Frameworks-Intern--Cosmos-Lab---2027_JR2029468?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply)