Performance Engineering Intern - jobs at Nvidia in China, Shanghai

for an exciting and fun role in our, which covers the areas over runtime, compiler optimization, code generation, CPU pipeline analysis and etc.

What you'll be doing:

• Work on feature enablement and optimization of open source Java virtual machine projects.

• Partner with both internal and communities to develop, valid and upstream patches.

• Work with global compiler, hardware and application teams to oversee improvements and problem resolutions.

• Drive and push architecture neutral and NV friendly solutions in the communities.

• Be flexible, with a variety of software development skills and extend the breadth and depth of knowledge.

What we need to see:

• Pursuing B.S. or higher degree in Computer Science/Engineering

• Excellent hands-on C++ programming skills

• Strong background in software engineering principles with a focus on crafting robust andmaintainable solutions to challenging problems

• Good communication and documentation skills and self-motivated

Ways to stand out from the crowd:

• Masters or PhD preferred, with compiler or language virtual machine experiences

• Background with Java and understand the key language features

• Familiar with Linux and shell programming

• Knowledge on computer architecture, ISA, assembly, and Arm64 is preferred

• Exposure to various ML techniques

What you’ll be doing:

• Develop algorithms to exercise various parts of the GPU pipeline to verify our performance metrics.

• Deeply dive into NVIDIA GPU architecture and software stack, develop new feature for NVIDIA GPU performance profiling tools.

• Write unit and integration tests to verify the functionality, performance, stability, resource usage of our products.

What we need to see:

• Pursuing a Master's degree major in CS/SE.

• Proficiency in C/C++, object oriented programming.

• Proficiency in written and spoken English.

Ways to stand out from the crowd:

• OpenGL, GLES, Direct3D, Vulkan, CUDA, OpenCL, console graphics APIs.

• Experience of driver development.

• Background with software development for embedded systems.

What you’ll be doing:

• Lead, mentor, and grow a high-performing team of applied research engineers focused on humanoid loco-manipulation and mobile manipulation.

• Drive the definition, planning, and execution of projects involving foundation models (GR00T, Cosmos), Isaac Lab, and Newton workflows.

• Guide the team in advancing robot learning technologies and synthetic data generation from human video datasets.

• Hands-on design, implementation, and deployment of novel algorithms for humanoid robot locomotion and manipulation in simulation and real-world environments.

• Ensure seamless integration of applied research outputs with NVIDIA’s advanced robotics platforms.

• Foster a culture of innovation and collaboration, supporting deliverables such as prototypes, open source software contributions, patents, and publications in top conferences and journals.

• Collaborate cross-functionally with product, hardware, and software teams to translate research into impactful products.

• Support career development, performance management, and recruitment of top talent.

What we need to see:

• Advanced degree (PhD or Master’s) in Computer Science, Robotics, or a related field.

• 2 years of experience on technical leadership or team management in robotics, autonomous driving, machine learning, or related domains.

• Strong hands-on programming skills in Python and C++; experience with deep learning frameworks (PyTorch, JAX, TensorFlow) and physics simulation tools (Isaac Sim/Lab, MuJoCo).

• Excellent communication, organizational, and interpersonal skills.

• Experience with large-scale model training on GPU clusters.

• Hands-on experience on robotics simulation, sim-to-real and real-to-sim transfer.

• 5+ overall years of experience working on robotics technologies.

Ways to stand out from the crowd:

• Leadership in projects involving foundation models for robotics, includingVision-Language-Action(VLA) or Vision-Language Models (VLM).

• Experience with learning from human video demonstrations and human-object reconstruction.

• Expertise in humanoid loco-manipulation, encompassing whole-body control, dexterous and bimanual manipulation, and locomotion.

• Advanced knowledge in robot learning and reasoning, including imitation and reinforcement learning.

• Experience generating synthetic data for robotics applications.

What you’ll be doing:

• Apply Machine Learning, Deep Learning techniques to overcome QA and Automation challenges for different NVIDIA Product lines

• Maintain and optimize new and existing automated tests with AI solution

• Design, implement and augment existing test frameworks with AI automation tools to reduce manual development efforts and make the current automation framework more productive and smart

• Work closely and collaboratively with other development and QA teams to gather automation requirements and review automation design

What we need to see:

• Pursuing Master or higher degree in computer science or electrical engineering with strong academics

• Good programming skills in python, proficient in applying OOP concepts and Data structures

• Good experience with using AI development tools for test plan creation, automation framework and test case development

• Have experience on using Cursor, MCP, Coderabbit for script generation/review

• Hands on experience in solving complex problems using AI technology or tools such as RAG, LLM/vLLM and AIGC

• Experience with large or complex applications and knowledge of code optimization and manipulation

• Demonstrates excellent communication skills, maintains a strong sense of initiative and motivation, and consistently upholds high standards for software quality

• Ability to work closely & collaboratively with other Development & QA teams especially across geographies

Ways to stand out from the crowd:

• Programming skills in C# is a plus

• Knowledge of Linux, Mac and Android is a plus

• Experience on Machine Learning Frameworks like Pytorch, Keras with TensorFlow, ONNX , and TensorRT is a plus

• Background of RL(Reinforce learning), or Meta-learning, or Life-long learning is a plus.

• Knowledge of SQL/NoSQL Database is plus

What you'll be doing:
• Work on research, design and implementation of software features that are beneficial to our customers to meet their performance targets and build unique values

• World model and nvidia cosmos related development and bug fixes

• Develop solutions for DNN models acceleration optimization and deployment.

What we need to see:
• Subject to arrangement to different works at any time to take on different tasks and challenges

• Self-motivated attitude and motivation to make things success, intent to learn

• BS/MS degree in Computer Science/EE or related

• Proven fundamentals in c++/python programming and SW design and debug skills

• Abundant knowledge of ML/DL techniques for Computer Vision and autonomous driving

• Believe that experimenting is the only way to find the truth, not argument and arbitary guess

• Located in Shanghai or willing to work in Shanghai

Ways to stand out from the crowd:
• DNN development and network acceleration is highly desired

• Familiarity with GPU computing/NVIDIA CUDA/NVIDIA TensorRT

What you will be doing:

• Collaborate closely with our AI/ML researchers to make their ML models more efficient leading to significant productivity improvements and cost savings

• Build tools, frameworks, and apply ML techniques to detect & analyze efficiency bottlenecks and deliver productivity improvements for our researchers

• Work with researchers working on a variety of innovative ML workloads across Robotics, Autonomous vehicles, LLM’s, Videos and more

• Collaborate across the engineering organizations to deliver efficiency in our usage of hardware, software, and infrastructure

• Proactively monitor fleet wide utilization patterns, analyze existing inefficiency patterns, or discover new patterns, and deliver scalable solutions to solve them

• Keep up to date with the most recent developments in AI/ML technologies, frameworks, and successful strategies, and advocate for their integration within the organization.

What we need to see:

• BS or similar background in Computer Science or related area (or equivalent experience)

• Minimum 8+ years of experience designing and operating large scale compute infrastructure

• Strong understanding of modern ML techniques and tools

• Experience investigating, and resolving, training & inference performance end to end

• Debugging and optimization experience with NSight Systems and NSight Compute

• Experience with debugging large-scale distributed training using NCCL

• Proficiency in programming & scripting languages such as Python, Go, Bash, as well as familiarity with cloud computing platforms (e.g., AWS, GCP, Azure) in addition to experience with parallel computing frameworks and paradigms.

• Dedication to ongoing learning and staying updated on new technologies and innovative methods in the AI/ML infrastructure sector.

• Excellent communication and collaboration skills, with the ability to work effectively with teams and individuals of different backgrounds

Ways to stand out from the crowd:

• Background with NVIDIA GPUs, CUDA Programming, NCCL and MLPerf benchmarking

• Experience with Machine Learning and Deep Learning concepts, algorithms and models

• Familiarity with InfiniBand with IBOP and RDMA

• Understanding of fast, distributed storage systems like Lustre and GPFS for AI/HPC workloads

• Familiarity with deep learning frameworks like PyTorch and TensorFlow