英伟达Computer Architecture Intern - LLM, 2026
任职要求
• Proven experience in software engineering, particularly in GPU programming and LLM inference. • Strong proficiency in programming languages such as Python, C++, and CUDA. • A solid understanding of deep learning frameworks and techniques. • Outstanding problem-solving skills and the ability to work collaboratively in a tea…
工作职责
• Develop and refine software solutions to expedite LLM SW stack (could be within inference/post train or pre-train phase) by harnessing the power of GPU technology. • Collaborate closely with a world-class team of engineers to implement and refine GPU-based algorithms. • Analyze and determine the most effective methods to improve performance, ensuring seamless execution across diverse computing environments. • Engage in both individual and team projects, contributing to NVIDIA's mission of leading the AI revolution. • Work in an empowering and inclusive environment to successfully implement groundbreaking AI solutions.
NVIDIA is developing processor and system architectures that accelerate deep learning and high-performance computing applications. We are looking for an intern deep learning system performance architect to join our AI performance modelling, analysis and optimization efforts. In this position, you will have a chance to work on DL performance modelling, analysis, and optimization on state-of-the-art hardware architectures for various LLM workloads. You will make your contributions to our dynamic technology focused company. What you’ll be doing: • Analyze state of the art DL networks (LLM etc.), identify and prototype performance opportunities to influence SW and Architecture team for NVIDIA's current and next gen inference products. • Develop analytical models for the state of the art deep learning networks and algorithm to innovate processor and system architectures design for performance and efficiency. • Specify hardware/software configurations and metrics to analyze performance, power, and accuracy in existing and future uni-processor and multiprocessor configurations. • Collaborate across the company to guide the direction of next-gen deep learning HW/SW by working with architecture, software, and product teams.
• Working directly with key application developers (especially LLM) to understand the current and future problems they are solving, creating and optimizing core parallel algorithms and data structures to provide the best solutions using GPUs, through both library development and direct contribution to the applications. This includes training and inference optimization for large language models, directly contributing to frameworks such as Megatron and TRTLLM, SGLang, vLLM... • Collaborating closely with the architecture, research, libraries, tools, and system software teams at NVIDIA to influence the design of next-generation architectures, software platforms, and programming models, including by investigating impact on application performance and developer productivity. • Engaging in deep optimization of high-performance operators, involving but not limited to CUDA deep optimization, instruction and compiler optimization. These optimizations will directly support customers or be integrated into products like cuDNN, cuBLAS, and CUTLASS...
We are now looking for a GeForce/ProViz Performance Engineer Intern! This position offers the chance to create a significant impact in a dynamic, technology focused company. As a member of the Performance Lab team, you will reach firsthand GPUs and optimize performance from designing stage till whole product lifetime, architectures to extend the state of the art in Gaming, Professional Visualization, Cloud Gaming, Data Center efficiency and performance. What you’ll be doing: • Identify, run graphics, studio and WinAI benchmarks across servers, PCs, workstations and laptops. • Compose competitive analysis reports for internal and external customers to position NVIDIA products appropriately using their evaluation. • Develop and maintain automation scripts for games/studio/WinAI performance and system monitoring data collection on Windows and Linux to speed up providing business and engineering insights. • Develop, implement and maintain tools to improve testing efficiency.