Principal Product Manager, AI Frameworks
NVIDIA· Product Management
Santa Clara, CA$240k - $379.5kfull-timePosted 7mo ago
About the role
Product Manager responsible for post-training and RL tools, SDKs, and libraries enabling model builders to optimize large-scale performance and resilience on NVIDIA GPUs, with a focus on post-training RL libraries, frameworks, and related software across training and deployment stages.
Responsibilities
- Create and optimize post-training RL libraries, frameworks to help researchers and production model builders
- Develop product strategy, roadmaps, and go-to-market plans
- Collaborate with internal and external customers to build product-based roadmaps for training/post training software
- Work with leadership to align with and drive company strategy
Qualifications
- Experience with design and scaling of training/post training and optimization software (ex. VeRL, Tunix, Nemo Framework, PyTorch distributed, torchtitan, etc.)
- Demonstrable knowledge of GenAI or machine learning concepts, particularly around model training, performance optimization, and software development and delivery
- Experience with large scale distributed systems
- BS or MS degree in Computer Science, Computer Engineering, or similar experience (or equivalent experience)
- 15+ years of technical product management, or similar, experience at a technology company
- Strong communication and interpersonal skills
Skills
Required:Post-training RL librariesDistributed systemsGenAI / ML concepts
Benefits
- Equity
- Benefits (see NVIDIA benefits)
- Base salary and compensation package including health, retirement, and other standard benefits as per company policy
Last updated Jul 3, 2026. Interested? Apply on the company site.
About NVIDIA
World Leader in AI Computing
NVIDIA invents the GPU and drives advances in AI, HPC, gaming, creative design, autonomous vehicles, and robotics.
AIautonomous vehiclesgaminggpugraphics cardhigh performance computingliquid-ai-recruiting-targetllm
More open roles at NVIDIA
- Senior Physical Verification Engineer4d agoEngineer, ASIC
- Senior System Software Engineer, Display Security4d agoEngineer, Sys SW
- Senior FPGA Prototyping Engineer - Hardware4d agoSanta Clara, CA · Engineer, ASIC
- Senior Product Engineer4d agoHybrid (Thailand) · Engineer, Product Development
- Senior Architect - Data Processing Systems, Analytics and Data Intelligence4d agoHybrid (Remote / Local) · Architect