Principal Product Manager, AI Frameworks jobs in United States
cer-icon
Apply on Employer Site
company-logo

NVIDIA · 1 month ago

Principal Product Manager, AI Frameworks

positionUS, CA, Santa Clara
timeFull-time
remoteOnsite
seniorityLead/Staff
money$240K/yr - $380K/yr
date15+ years exp

NVIDIA is a leading technology company focused on AI and GPU solutions. As a Principal Product Manager for AI Frameworks, you will be responsible for building tools and libraries that enhance performance and resilience for model builders using NVIDIA GPUs.

AI InfrastructureArtificial Intelligence (AI)Consumer ElectronicsFoundational AIGPUHardwareSoftwareVirtual Reality
check
Growth Opportunities
check
H1B Sponsor Likelynote

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

Qualification

AI frameworksProduct managementMachine learningLarge scale systemsGenAI knowledgeOpen Source experienceGPU architecture knowledgeCommunicationInterpersonal skills

Required

Experience with training/post training and optimization software (ex. 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
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

Preferred

Experience leading GenAI/RecSys research to production at scale
Working on Open Source & Github-first developer products with deep customer interactions
Knowledge of GPU architecture, HW/SW co-design, and performance profiling

Benefits

Equity
Benefits

Company

NVIDIA

twittertwittertwitter
Glassdoor4.6
company-logo
NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.
dateFounded in 1993
location
Santa Clara, California, USA
people-size10001+ employees
web-link
https://www.nvidia.com

H1B Sponsorship

NVIDIA has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1877)
2024 (1355)
2023 (976)
2022 (835)
2021 (601)
2020 (529)

Funding

Current Stage
Public Company
Total Funding
$4.09B
Key Investors
ARPA-EARK Investment ManagementSoftBank Vision Fund
2023-05-09Grant· $5M
2022-08-09Post Ipo Equity· $65M
2021-02-18Post Ipo Equity

Leadership Team

leader-logo
Jensen Huang
Founder and CEO
linkedin
leader-logo
Michael Kagan
Chief Technology Officer
linkedin

Recent News

GlobeNewswire
NVIDIA and Lilly Announce Co-Innovation AI Lab to Reinvent Drug Discovery in the Age of AI
2026-01-13
The Motley Fool
From Training AI to Designing Chips: Nvidia's Next Platform Play
2026-01-13
PR Newswire
Freenome Announces Expanded Artificial Intelligence and Deep Learning Initiatives Accelerated by NVIDIA, to Advance Personalized Multi-Cancer Detection
2026-01-13
Company data provided by crunchbase