NVIDIA
Principal Product Manager, AI Frameworks
$240K–$380K
US, CA, Santa Clara · Principal · Full-time · Posted Jan 9, 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
Leads product strategy and development for AI training and post-training tools, SDKs, and libraries that help model builders achieve performance and resilience on NVIDIA GPUs.
What you'll do
- Create products for researchers and production model builders; develop product strategy, roadmaps, and go-to-market plans.
- Work with internal and external customers to shape roadmaps for training and post-training software.
- Align product work with company strategy alongside NVIDIA leadership.
What they ask for
- Experience with training/post-training and optimization software.
- 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 GenAI/RecSys research to production at scale.
- Open Source and Github-first developer product experience, deep customer interactions, or knowledge of GPU architecture, HW/SW co-design, and performance profiling.
Pay
240,000 USD - 379,500 USD base salary
Tools
PyTorch distributed, torchtitan, VeRL, Nemo Framework
AI focus
Builds AI training and post-training products and software for model builders; supports large-scale performance, resilience, and research-to-production on NVIDIA GPUs.
Model-written summary · the original posting is authoritative