Principal Product Manager, AI Frameworks

About the role

Nvidia seeks a Product Manager for AI Platform training who will develop tools, SDKs, and libraries to enhance model builders' large-scale performance, resilience, and experience on NVIDIA GPUs. This role involves working closely with customers and internal stakeholders to shape product strategies, roadmaps, and go-to-market plans.

Responsibilities

• Create products 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

Requirements

• Experience with training/post training and optimization software (e.g., PyTorch distributed, torchtitan, VeRL, Nemo Framework, etc.)
• Demonstrable knowledge of GenAI or machine learning concepts, particularly around model training, performance optimization, and software development and delivery
• 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

Qualifications

• Experience leading GenAI/RecSys research to production at scale
• Experience with large scale distributed systems

Skills

• Experience with Open Source & Github-first developer products
• Deep customer interactions
• Knowledge of GPU architecture, HW/SW co-design, and performance profiling

Benefits

Base salary range: $240,000 - $379,500 USD. Eligible for equity and benefits.

Pay

Salary will be determined based on location, experience, and the pay of employees in similar positions.

Schedule

Open until January 13, 2026.