AI Developer Technology Intern, Robotics - 2027
At a glance
Junior AI Engineering role at NVIDIA. China, Shanghai · full-time.
Pay not stated AI Engineering salary
aimlroles summary, based on the employer's posting.
What you'll do
- Build and tune training workflows for robotics foundation models, including vision-language-action and world models, on NVIDIA GPU clusters
- Measure inference speed on NVIDIA platforms, identify bottlenecks, and improve model performance
- Use deep learning methods to tackle practical robotics problems
- Work with architecture, research, libraries, tools, and system software teams on future platforms and programming models
- Create robotics AI examples, reference code, and developer documentation, then share results with senior DevTech engineers
What you bring
- You are working toward a master's or PhD in computer science, robotics, AI/ML, electrical engineering, or a related technical area
- Strong Python skills and experience with deep learning frameworks such as PyTorch
- You understand transformer designs and training large-scale models
- You have basic grounding in robotics, such as perception, control, kinematics, or simulation
Who this fits
This internship suits a graduate student interested in robotics foundation models, GPU performance, and hands-on ML software development. You should be comfortable taking on open-ended technical problems and discussing your work with senior engineers.
From the employer
Today, NVIDIA is tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. We are the DevTech team serving the robotics ecosystem in China. As NVIDIA DevTechs, we are defined as experts in application optimization on NVIDIA platforms — through code optimization, new application architectures, and cutting-edge DL/ML algorithms. Our mission spans performance analysis & modeling, applying modern DL/ML techniques to real-world problems, deep-diving on GPU architecture, building samples/libraries/SDKs, and collaborating with architecture, software engineering, and product management to improve NVIDIA’s current and future platforms. We are looking for a passionate intern to work on Robotics Foundation Model training and inference — powering the next generation of intelligent robotic systems on NVIDIA’s platform.
What You’ll Be Doing:
Develop and optimize training pipelines for robotics foundation models (e.g., vision-language-action models, world models) on NVIDIA GPU clusters
Profile, analyze, and optimize model inference performance on NVIDIA platforms
Apply modern DL/ML techniques to real-world robotics problems
Collaborate closely with NVIDIA architecture, research, libraries, tools, and system software teams to influence the design of next-generation architectures, software platforms, and programming models
Build samples, reference implementations, and documentation to educate developers on NVIDIA robotics AI technologies
Discuss your approach and results together with senior DevTech engineers
What We Need to See:
Currently pursuing a Master’s or PhD degree in Computer Science, Robotics, AI/ML, Electrical Engineering, or a related technical field
Strong programming skills in Python; familiarity with C/C++ and CUDA is a plus
Experience with deep learning frameworks (PyTorch,)
Understanding of transformer architectures and large-scale model training
Basic knowledge of robotics concepts (kinematics, control, perception, or simulation)
Curious, self-motivated, and excited about solving open-ended challenges at NVIDIA
Ways to Stand Out from the Crowd:
Experience with distributed training (DeepSpeed, Megatron-LM, FSDP, NeMo)
Experience with inference optimization (SGlang, vLLM)
Familiarity with NVIDIA robotics platforms (Isaac Sim, Isaac Lab, GR00T)
Publications or project experience in embodied AI or robotics foundation models
Ambitious to grow and learn about building ML applications, optimization, and software engineering