MNC InsiderMNC Insider
NVIDIA logo

Principal Product Manager, AI Frameworks

NVIDIA

Principal Product Manager, AI Frameworks

full-timePosted: Jul 28, 2026Updated: Aug 27, 2026Santa Clara, CA, US

Job Description

At NVIDIA, we meet customers where they are on their AI journey on our GPUs - this means we build best in class frameworks in OSS and support a robust ecosystem of other OSS frameworks. NVIDIA's PyTorch Compilers team builds and upstreams to the stack that sits between PyTorch and NVIDIA hardware — spanning torch.compile, emerging compiler substrates, and the agent-native optimization infrastructure being built for the next era of accelerated computing. This role will build and direct product strategy across the full arc. It involves shipping the latest hardware features in torch.compile today. It also includes developing the canonical shared representation across NVIDIA's compiler and runtime ecosystem. Additionally, it focuses on crafting how agents will engage in deep learning performance work in the future. We are looking for someone who understands compilers and can operate at the intersection of systems architecture, framework engineering, and customer-facing product strategy working directly with engineering leadership and NVIDIA's most sophisticated external customers — including frontier model labs, inference/training and RL framework teams such as vLLM, SGLang, torchtitan, megatron-core, and hardware co-design programs. As NVIDIA Product Managers, we partner with NVIDIA leaders to define clear product strategy, and marketing team teams to build go-to-market plans. The Product Management organization at NVIDIA is a flexible, strong, and impactful group focusing on enabling deep learning across all GPU use cases and providing great products for our users. We seek an individual with a rare blend of product skills, technical depth, and passion to join our team. Does that sounds familiar? If so, we would love to hear from you!What you'll be doing:Own the strategy for NVIDIA's PyTorch compiler portfolio, including:torch.compile — maintain and evolve NVIDIA's upstream PyTorch path, drive HW support, resolve customer issues across dynamic shapes, kernel performance, and compile overheaddefine the roadmap and go-to-market for NVIDIA's shared representation layer across frameworks, compilers, kernel libraries, and runtimesAgent-native compiler workflows — shape the product vision for how agentic systems will participate in optimization tasksInter-kernel optimization features — build product requirements for capabilities like megakernels and ensure they ship with demonstrated real-world valueLead product strategy across the roadmap:Define Now/Next/Later priorities in close partnership with engineering leadsTranslate ecosystem signals (vLLM RFCs, SGLang CUDA Graph patterns, TorchTitan GraphTrainer plans, Meta's upstream priorities) into prioritized product decisionsEngage directly with customers and partners:Represent NVIDIA at PyTorch contributor and ecosystem forums;Define success metrics and release criteria:Establish performance gates and adoption milestones Set bar for what "proven" means for new optimizations before committing to roadmapWhat we need to see:15+ years in technical product management, with 5 years owning a compiler, runtime, or low-level systems product at scaleBS or MS degree in Computer Science, Electrical Engineering, a related technical field, or equivalent experience.Experience with OSS-first products and upstream contribution strategyTrack record of shipping and driving adoption for developer efficiency and performance oriented infrastructure products Understands how inference frameworks use compiler technology — where they adopt torch.compile, where they go around it, and whyUnderstands how new hardware features create new compiler requirementsCan write clear, defensible strategy documents and knows how to scope an early-access release:Strong instinct for where to concentrate investment vs. spread it Ways to stand out from the crowd:Deep understanding of the PyTorch compiler stack and how to influence strategy in this ecosystem You've worked on how agents participate in systems-level optimization workflowsFamiliarity with MLIR-based compiler infrastructure and how it maps to NVIDIA hardware primitivesCan reason about inter-kernel optimization tradeoffs to define the right bar for "proven"Comfortable reading kernel performance profiles and debug how torch.compile can help any model #LI-HybridYour base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 240,000 USD - 379,500 USD.You will also be eligible for equity and benefits.Applications for this job will be accepted at least until July 27, 2026.This posting is for an existing vacancy. NVIDIA uses AI tools in its recruiting processes.NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, 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.

Locations

  • Santa Clara, CA, US

Responsibilities

  • Own the strategy for NVIDIA's PyTorch compiler portfolio, including:
  • torch.compile — maintain and evolve NVIDIA's upstream PyTorch path, drive HW support, resolve customer issues across dynamic shapes, kernel performance, and compile overhead
  • define the roadmap and go-to-market for NVIDIA's shared representation layer across frameworks, compilers, kernel libraries, and runtimes
  • Agent-native compiler workflows — shape the product vision for how agentic systems will participate in optimization tasks
  • Inter-kernel optimization features — build product requirements for capabilities like megakernels and ensure they ship with demonstrated real-world value
  • torch.compile — maintain and evolve NVIDIA's upstream PyTorch path, drive HW support, resolve customer issues across dynamic shapes, kernel performance, and compile overhead
  • define the roadmap and go-to-market for NVIDIA's shared representation layer across frameworks, compilers, kernel libraries, and runtimes
  • Agent-native compiler workflows — shape the product vision for how agentic systems will participate in optimization tasks
  • Inter-kernel optimization features — build product requirements for capabilities like megakernels and ensure they ship with demonstrated real-world value
  • Lead product strategy across the roadmap:
  • Define Now/Next/Later priorities in close partnership with engineering leads
  • Translate ecosystem signals (vLLM RFCs, SGLang CUDA Graph patterns, TorchTitan GraphTrainer plans, Meta's upstream priorities) into prioritized product decisions
  • Define Now/Next/Later priorities in close partnership with engineering leads
  • Translate ecosystem signals (vLLM RFCs, SGLang CUDA Graph patterns, TorchTitan GraphTrainer plans, Meta's upstream priorities) into prioritized product decisions
  • Engage directly with customers and partners:
  • Represent NVIDIA at PyTorch contributor and ecosystem forums;
  • Represent NVIDIA at PyTorch contributor and ecosystem forums;
  • Define success metrics and release criteria:
  • Establish performance gates and adoption milestones
  • Set bar for what "proven" means for new optimizations before committing to roadmap
  • Establish performance gates and adoption milestones
  • Set bar for what "proven" means for new optimizations before committing to roadmap
  • This posting is for an existing vacancy.
  • NVIDIA uses AI tools in its recruiting processes.

Target Your Resume for "Principal Product Manager, AI Frameworks" , NVIDIA

Get personalized recommendations to optimize your resume specifically for Principal Product Manager, AI Frameworks. Takes only 15 seconds!

AI-powered keyword optimization
Skills matching & gap analysis
Experience alignment suggestions

Check Your ATS Score for "Principal Product Manager, AI Frameworks" , NVIDIA

Find out how well your resume matches this job's requirements. Get comprehensive analysis including ATS compatibility, keyword matching, skill gaps, and personalized recommendations.

ATS compatibility check
Keyword optimization analysis
Skill matching & gap identification
Format & readability score

Tags & Categories

GeneralGeneral

Answer 10 quick questions to check your fit for Principal Product Manager, AI Frameworks @ NVIDIA.

Quiz Challenge
10 Questions
~2 Minutes
Instant Score

Related Books and Jobs

No related jobs found at the moment.