AI Developer Technology Engineer
At a glance
Mid-level 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
- Help China-based robotics teams shape and scale embodied AI workloads using NVIDIA platforms
- Extend Isaac Lab, Megatron Core and VLM/VLA/WA inference frameworks to sustain performance at scale
- Tune GPU physics simulations and robot-model training and inference pipelines, then feed findings into product work
- Build sample workflows for imitation learning, reinforcement learning, and VLA, WA or VLM model training and inference
- Present technical results to developers, collaborate with engineering groups, and represent the team at regional events
What you bring
- A computer science, robotics or engineering degree, or equivalent experience
- At least three years of work experience, plus skills in C++, CUDA, Python and Linux
- Experience using a physics simulator such as MuJoCo, Isaac Sim or PyBullet, or another listed simulator
- Background in embodied AI or physics simulation work at a humanoid robotics company
Who this fits
This role suits someone with practical experience in embodied AI and robot simulation who wants to develop and optimize machine learning applications. You’ll work with developers and researchers in China and take part in regional events, workshops and customer engagements.
From the employer
NVIDIA is the engine of modern AI. For over 25 years our work in accelerated computing has redefined computer graphics, autonomous machines, and now physical AI. We are looking for an AI Developer Technology Engineer to join our DevTech team in China. this role you will work directly with the developers and researchers building the next generation of robots, helping them get the most out of NVIDIA GPUs across simulation, training, and deployment. You will be part of an exceptionally experienced, globally distributed team dedicated to pushing the boundaries of embodied AI.
What you’ll be doing
- Partner with leading robotics developers and research teams in China to design, optimize, and scale embodied AI workloads on NVIDIA platforms
- Develop on and extend frameworks such as Isaac Lab, Megatron Core and VLM/VLA/WA inference framework, ensuring flawless performance at scale
- Profile and optimize GPU-based physics simulation, robot foundation model training and inference pipelines; drive the findings back into our products
- Build reference workflows and samples for imitation learning, reinforcement learning, and VLA/WA/VLM model training/inference
- Discuss your approach and results with NVIDIA engineers and external developers to continuously improve processes
- Collaborate closely with architecture, research, libraries, tools, and system software teams to invent and develop next-generation architectures, software platforms, and programming models
- Represent NVIDIA technically at developer events, workshops, and customer engagements in the region
What we need to see:
- Bachelor's degree in Computer Science, Robotics, Engineering, or a related field, or equivalent experience.
- 3+ working year experience.
- Experience with C++, CUDA, Python, and Linux.
- Proven experience with one or more physics simulators such as MuJoCo, Isaac Sim, PyBullet, Drake, or Gazebo.
- Strong communication skills and the ambition to grow and learn about building machine learning applications, optimization, and software engineering.
- Prior involvement with embodied AI or a history at humanoid robotics firms with a focus on physics simulation.
Widely considered to be one of the technology world’s most desirable employers, NVIDIA offers highly competitive salaries and a comprehensive benefits package.Are you a creative and autonomous computer scientist with a real passion for parallel computing? If so, we want to hear from you.