Logo
Nvidia Corporation

Senior Deep Learning Manager, LLM Inference

Nvidia Corporation, Santa Clara, California, us, 95053

Save Job

NVIDIA is at the forefront of the generative AI revolution. The Inference Benchmarking (IB) team focuses on advanced inference server performance for Large Language Models (LLMs). If you are passionate about pushing the boundaries of GPU hardware and software performance and understand terms like pre-fill phase, generation phase, paged attention, MoE, Tensor Parallel, Llama, GPT-OSS, and HuggingFace, this could be a great role for you.

Overview NVIDIA is at the forefront of the generative AI revolution. The Inference Benchmarking (IB) team specifically focuses on advanced inference server performance for Large Language Models (LLMs).

Responsibilities

You will be responsible for managing a team that characterizes the latest LLMs and inference servers like TensorRT-LLM, vLLM, and SGLang to ensure that NVIDIA maintains its leadership position.

Join forces with the performance marketing team to build engaging content, including blog posts and other written materials, that highlight TensorRT-LLM's outstanding achievements.

Collaborate with engineers from AI startup companies to debug and establish standard methodologies.

Profile GPU kernel-level performance to identify hardware and software optimization opportunities.

Develop profiling and analysis software tools that can keep up with the rapid pace of network scaling.

Contribute to deep learning software projects, such as PyTorch, TRT-LLM, vLLM, and SGLang to drive advancements in the field.

Verify that TRT-LLM's performance meets expectations for new GPU product launches.

Collaborate across the company to guide the direction of inference serving, working with software, research, and product teams to ensure world-class performance.

Qualifications

Master's or PhD degree in Computer Science, Computer Engineering, or related fields, or equivalent experience.

10+ overall years of software development experience and at least 3 years of management experience.

Detailed knowledge of deep learning inference serving, PyTorch programming, profiling, and compiler optimizations.

Proficiency in Python and C++ programming languages and familiarity with CUDA.

Experience with LLMs and their performance challenges and opportunities.

Solid understanding of CPU and GPU microarchitecture and performance characteristics.

Experience with complex software projects like frameworks, compilers, or operating systems.

Good written and verbal communication skills and the ability to work independently and collaboratively in a fast-paced environment.

Ways to stand out from the crowd

Demonstrate a drive to continuously improve software and hardware performance.

Showcase examples of novel use cases for agentic AI tools in the workplace.

Experience with database and visualization tools like D3.js will set you apart.

Compensation and benefits Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 272,000 USD - 425,500 USD. You will also be eligible for equity and benefits.

Application timeline Applications for this job will be accepted at least until September 6, 2025.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. We do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.

#J-18808-Ljbffr