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

NVIDIA · 2 days ago

Principal Product Manager, AI Frameworks

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

NVIDIA is a leading company in AI technology, and they are seeking a Principal Product Manager for AI Frameworks. The role involves building tools, SDKs, and libraries for model builders to optimize performance on NVIDIA GPUs, while also collaborating with customers and defining product strategies.

Artificial Intelligence (AI)Consumer ElectronicsGPUHardwareSoftwareVirtual Reality
check
Growth Opportunities
check
H1B Sponsor Likelynote
Hiring Manager
Dayna W.
linkedin

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

MarketScreener
Nvidia CEO Huang touts automotive AI at CES talk as competition mounts
2026-01-06
MarketScreener
Nvidia: Rubin-based products will be available from second half of 2026
2026-01-06
GlobeNewswire
NVIDIA Announces Alpamayo Family of Open-Source AI Models and Tools to Accelerate Safe, Reasoning-Based Autonomous Vehicle Development
2026-01-06
Company data provided by crunchbase