Principal Product Manager, AI Frameworks

The NVIDIA PyTorch Compilers team is seeking a Product Manager to lead the strategy for NVIDIA's PyTorch compiler portfolio. This role involves owning the strategy for torch.compile, defining the roadmap and go-to-market for the shared representation layer, shaping the product vision for agent-native compiler workflows, and leading product strategy across the entire roadmap.

About the role

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

Responsibilities

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
Engage Directly With Customers And Partners
Represent NVIDIA at PyTorch contributor and ecosystem forums; Define Success Metrics And Release Criteria
Set bar for what "proven" means for new optimizations before committing to roadmap

Requirements

• 15+ years in technical product management, with 5 years owning a compiler, runtime, or low-level systems product at scale
• Bachelor's or Master's degree in Computer Science, Electrical Engineering, a related technical field, or equivalent experience
• Experience with OSS-first products and upstream contribution strategy
• Track 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 why
• Understands how new hardware features create new compiler requirements
• Can write clear, defensible strategy documents and knows how to scope an early-access release
• Strong instinct for where to concentrate investment vs. spread it

Qualifications

• Deep understanding of the PyTorch compiler stack and how to influence strategy in this ecosystem
• Experience working on how agents participate in systems-level optimization workflows
• Familiarity with MLIR-based compiler infrastructure and how it maps to NVIDIA hardware primitives
• Reasoning about inter-kernel optimization tradeoffs to define the right bar for "proven"
• Comfortable reading kernel performance profiles and debugging how torch.compile can help any model

Your 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 - $379,500. 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.