NVIDIA
Principal Product Manager, AI Frameworks
$240K–$380K
US, CA, Santa Clara · Principal · Full-time · Posted Jan 26, 2026
- Experience
- 15+ years of technical product management, or similar, experience at a technology company.
- Education
- BS or MS degree in Computer Science, Computer Engineering, or similar experience (or equivalent experience).
About the job
Builds and leads product strategy for post-training reinforcement-learning tools, frameworks, and libraries that help researchers and production model builders achieve performance and resilience on NVIDIA GPUs.
What you'll do
- Create and optimize post-training RL libraries and frameworks.
- Develop product strategy, roadmaps, and go-to-market plans.
- Work with customers and company leadership to shape software roadmaps and align with company strategy.
What they ask for
- Experience designing and scaling training/post-training and optimization software.
- Demonstrable knowledge of GenAI or machine learning, especially model training, performance optimization, and software development and delivery.
- Experience with large-scale distributed systems.
- Strong communication and interpersonal skills.
Nice to have
- Experience leading RL from research to production at scale.
- Experience with open-source, GitHub-first developer products and deep customer interactions; knowledge of GPU architecture, HW/SW co-design, and performance profiling.
Pay
240,000 USD - 379,500 USD base salary
Tools
VeRL, Tunix, Nemo Framework, PyTorch distributed, torchtitan
AI focus
Build tools, SDKs, libraries, and frameworks for AI model post-training and reinforcement learning, supporting large-scale performance and resilience on NVIDIA GPUs.
Model-written summary · the original posting is authoritative