NVIDIA

Software Engineer, Robotics - Isaac Lab

China, Shanghai Full time

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. Doing what’s never been done before takes vision, innovation, and the world’s best talent. As an NVIDIAN, you’ll be immersed in a diverse, encouraging environment where everyone is inspired to do their best work. Come join the team and see how we can make a lasting impact on the world.We are seeking a software engineer to join the Isaac Lab team and propel our flagship platform for robot learning to new heights. Our mission is to become the industry's leading tool, redefining how autonomous systems are trained and shaping the future of robotics and AI.

What you’ll be doing:

  • Design, build, and maintain CI/CD pipelines, Docker images, and release workflows for large-scale robotics simulation and learning systems.

  • Automate workflows and scale simulation, training, and evaluation in cloud and local environments.

  • Perform extensive CPU and GPU benchmarking, profiling, and regression analysis to ensure best-in-class system performance, stability, and reproducibility.

  • Own system-level performance metrics (simulation step time, rendering throughput, training iteration speed) and proactively identify bottlenecks.

  • Support issue triage by creating reproducible environments and performance test cases.

  • Collaborate with research and engineering teams across NVIDIA to ensure reliable integration of simulation, learning, and rendering components.

What we need to see:

  • BS, MS, or PhD degree in Computer Science, or equivalent experience.

  • Strong software engineering skills in Python; familiarity with the deep-learning software stack (PyTorch preferred) and NVIDIA Warp.

  • 3 years experience with infrastructure for ML or simulation systems (CI/CD, Docker, distributed or GPU workloads).

  • Hands-on experience profiling and optimizing complex systems.

Ways to stand out from the crowd:

  • Prior experience with Isaac Sim, Isaac Lab, Isaac Gym, or Mujoco.

  • Experience maintaining CI or infrastructure for robotics, simulation, or ML platforms; and diagnosing performance regressions across physics, rendering, and learning pipelines.

  • Contributions to open-source infrastructure or tooling.

With highly competitive salaries and a comprehensive benefits package, NVIDIA is widely considered to be one of the technology industry's most desirable employers. We have some of the most forward-thinking and hardworking people in the world working with us and our product lines are growing fast in some of the hottest fields such as virtual reality, deep learning, artificial intelligence, and more. Come join us at NVIDIA!