Performance Engineering Intern, Deep Learning and HPC - 2027
Salary not listed • China, Shanghai • Internship
Posted 2d ago
About the job
We are seeking a Performance Engineering Intern to support our investments in performance testing of NVIDIA's datacenter products and applications. You will join the global Performance Lab team, enhancing our ability to accurately benchmark state-of-the-art AI applications and products. This role involves developing infrastructures and solutions to improve data gathering through automation and designing efficient testing processes for diverse applications and hardware. The data collected will inform marketing/sales collaterals and engineering studies for future products. You will collaborate with multi-functional teams in a dynamic environment with multiple concurrent projects and shifting priorities.
Responsibilities
- Benchmark, profile, and analyze AI and HPC workloads on NVIDIA supercomputers and distributed systems, including LLM and VLM training and inference, agentic workloads, world models, and Physical AI for robotics.
- Aggregate and produce written reports with testing data for internal sales, marketing, SW, and HW teams.
- Build benchmark automation using AI agents and agentic coding tools, ensuring reliability and reproducibility.
- Collaborate with internal teams to debug and resolve performance issues.
- Set up and configure systems with appropriate hardware and software for running benchmarks.
Requirements
- Currently pursuing a bachelor's degree (or higher) in Computer Science, Electrical Engineering, or a related field.
- Experienced in programming and debugging with scripting languages like Python or Unix shell.
- Hands-on experience with Linux-based systems.
- Familiarity with container platforms such as Docker.
- Fluent with AI coding assistants and agentic workflows, with the ability to verify their output.
- Hands-on experience with inference frameworks like TensorRT-LLM, vLLM, or SGLang, or experience in AI model inference deployment and training.
- Strong data analysis skills and the ability to summarize findings in written reports.
- Strong English verbal and written communication skills.