Skip to main content
careers

CAREERS AT NVIDIA

Deep Learning Performance Architect Intern - 2025 page is loaded

Deep Learning Performance Architect Intern - 2025

locations
China, Shanghai
China, Beijing
time type
Full time
posted on
Posted 27 Days Ago
job requisition id
JR1998117

We are looking for a first-class Deep Learning Performance architect to join in us to shape the performance analysis infrastructures for GPUs. We build cutting-edge analysis tools and visualization frameworks that empower engineers to optimize GPU performance for Deep Learning and HPC workloads—spanning pre-silicon architectural exploration to post-silicon validation and optimization. Your work will directly shape the tools that define how NVIDIA GPUs are analyzed, tuned, and scaled for next-gen AI systems, and impact the next-gen GPUs architectures.

What you’ll be doing:

  • Architect Performance Tooling: Develop infrastructure tools/libraries for GPU performance analysis, visualization, and automated workflows used across GPU SW/HW development life cycle

  • Unlock Architectural Insights: Analyze GPU workloads to identify bottlenecks and define new hardware profiling features that enhance perf debug and profiling capabilities.

  • AI-Powered Automation: Build AI/ML-driven tools to automate performance analysis, generate perf optimization guidance, and improve user experience of profiling infrastructure.

  • Cross-Stack Collaboration: Partner with kernel developers, system software teams, and hardware architects to co-design performance-centric solutions.

  • End-to-End Optimization: Create benchmarks to validate performance improvements across AI/HPC workloads and present actionable insights.

What we need to see: 

  • BS/MS+ in relevant discipline (CS, EE, Math)

  • Proficiency in C/C++ (performance-critical coding) and Python (automation/scripting, and AI/ML frameworks)

  • Strong grasp of computer architecture (pipelines, memory hierarchies) and Operating System fundamentals

  • Understand machine learning and data analysis basics, LLM techniques such as prompt engineering, fine-tuning, vector databases

  • Experience with performance modeling, architecture simulation, profiling, and analysis.

  • Self-starter who thrives in dynamic environments and manages competing priorities effectively.

Ways to stand out from the crowd:

  • Experience with developing HW performance debugging and analysis tools

  • Familiar with System Software Stack(like CUDA Driver), CUDA kernel optimization and understand GPU architecture

  • Familiarity with GPU performance profiling tools like Nsight System, Nsight Compute

  • Practical experience or projects demonstrating LLM-based code generation, automated data analysis, or workflow assistants. Prior experience with agentic LLM frameworks like Langchain and LLamaIndex.

  • Full-Stack Versatility: Skills in JavaScript, SQL, or UI/UX design for tool interfaces.

Similar Jobs (5)

locations
China, Shanghai
time type
Full time
posted on
Posted 30 Days Ago
locations
2 Locations
time type
Full time
posted on
Posted 30+ Days Ago
locations
China, Shanghai
time type
Full time
posted on
Posted 16 Days Ago

About Us

Logo

NVIDIA is the world leader in accelerated computing.

 

NVIDIA pioneered accelerated computing to tackle challenges no one else can solve. Our work in AI and digital twins is transforming the world's largest industries and profoundly impacting society.

 

Learn more about NVIDIA.

Follow Us
© 2025 Workday, Inc. All rights reserved.