3 days ago
Shanghai, China or Beijing, ChinaMid Level
Responsibilities
- Partner with robotics developers and research teams in China to design, optimize, and scale embodied AI workloads on NVIDIA platforms.
- Develop and extend Isaac Lab, Megatron Core, and VLM/VLA/WA inference frameworks.
- Profile and optimize GPU-based physics simulation, robot foundation-model training, and inference pipelines.
- Build reference workflows and samples for imitation learning, reinforcement learning, and VLA/WA/VLM model training and inference.
- Collaborate with architecture, research, libraries, tools, and system software teams on software platforms and programming models.
- Present technical approaches and results to NVIDIA engineers, external developers, and regional developer or customer audiences.
Requirements
- Bachelor's degree in Computer Science, Robotics, Engineering, or a related field, or equivalent experience.
- At least 3 years of working experience.
- Experience with C++, CUDA, Python, and Linux.
- Experience with one or more physics simulators, such as MuJoCo, Isaac Sim, PyBullet, Drake, or Gazebo.
- Strong communication skills and interest in machine learning applications, optimization, and software engineering.
- Prior involvement with embodied AI or experience at humanoid robotics firms focused on physics simulation.
Benefits
- NVIDIA offers competitive salaries and a comprehensive benefits package.
- The role is based in China and includes participation in developer events, workshops, and customer engagements in the region.
Categories
ML EngineeringRobotics
About Nvidia
Nvidia designs and sells GPUs and accelerated computing platforms for data centers, AI/ML, graphics, gaming, and automotive, monetizing through hardware, software platforms (CUDA, AI frameworks), and systems like DGX and networking. Customers include cloud providers, enterprises, researchers, and OEMs. Founded in 1993 and headquartered in Santa Clara, it is a public company traded on NASDAQ under NVDA.