Deep Learning Architect, LLM Inference - College Grad

Deep Learning Architect, LLM Inference

NVIDIA is at the forefront of the generative AI revolution. The Inference Benchmarking (IB) team specifically focuses on inference server performance optimization for Large Language Models (LLMs). If you’re passionate about pushing the boundaries of GPU hardware and software performance and understand terms like disaggregated serving, data parallel attention, MoE, Qwen
3.5, Deep Seek, GPT‑OSS, then this is a great role for you!

• You will do workload characterization of the latest LLMs and inference servers like vLLM, SGLang and TRT-LLM to ensure NVIDIA maintains its leadership position.
• Join forces with the performance marketing team to build engaging content, including blog posts and updates to Inference
  
  X to highlight NVIDIA’s outstanding inference achievements.
• Collaborate with engineers from AI startup companies to establish standard benchmarking methodologies.
• Develop a constantly evolving inference performance data results website.
• Invent E2E profiling and analysis tools that you will use to keep up with the rapid pace of Generative AI.
• Contribute to deep learning software projects, such as PyTorch, TRT-LLM, vLLM, and SGLang to drive advancements in the field.
• Verify that new GPU product launches produce industry leading performance.
• Collaborate across the company to guide the direction of inference serving, working with software, research, and product teams to ensure best‑in‑class performance.
• Use the latest coding agents and inference technology to improve team efficiency.

• Master’s or PhD degree in Computer Science, Computer Engineering, related fields, or equivalent experience.
• Relevant software development experience.
• Detailed knowledge of deep learning inference serving, PyTorch programming, profiling, and compiler optimizations.
• Experience developing client‑server LLM applications with OpenAI API or MCP and identifying performance bottlenecks.
• Solid understanding of CPU and GPU microarchitecture and performance characteristics.
• Experience with complex software projects like frameworks, compilers, or operating systems.
• Demonstrated proficiency with the latest AI coding agents like Claude Code, Codex, and Cursor.
• Excellent written and verbal communication skills and the ability to work independently and collaboratively in a fast‑paced environment.

• 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 databases and visualization tools will set you apart.

NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have a team of highly skilled and motivated individuals who excel in their work. If you have a proactive and independent approach, we want to hear from you!

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 124,000 USD – 195,500 USD for Level 2, and 152,000 USD – 241,500 USD for Level 3. You will also be eligible for equity and benefits.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal‑opportunity employer. We do not discriminate 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.