Principal Product Manager, AI Frameworks
$240,000–$379,500 year
HybridSanta Clara, California, United States
Job Summary
Own the strategy for NVIDIA's PyTorch compiler portfolio, including torch.compile maintenance, upstream path evolution, and hardware support. Define the roadmap and go-to-market for the shared representation layer across frameworks, compilers, and runtimes. Shape product vision for agent-native compiler workflows and inter-kernel optimization features like megakernels. Lead product strategy by translating ecosystem signals into prioritized decisions and establishing performance gates for new optimizations. Engage directly with customers and partners at PyTorch forums while defining success metrics and release criteria.
Required Qualifications
- 15+ years in technical product management
- 5 years owning a compiler, runtime, or low-level systems product at scale
- BS or MS 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
- 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 workflows
- Familiarity with MLIR-based compiler infrastructure and how it maps to NVIDIA hardware primitives
- Can 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
Desired Qualifications
- BS or MS degree in Computer Science, Electrical Engineering, a related technical field, or equivalent experience
Hiring someone like this?
Get your role in front of qualified candidates on Sorce.