← All roles

Principal Product Manager, AI Frameworks

NVIDIA· Product Management
Apply on company site
Santa Clara, CA$240k - $379.5kFull-timePosted 7mo ago

About the role

Lead product manager for AI Platform training at Nvidia, responsible for building tools, SDKs, and libraries to optimize large-scale model training and deployment on NVIDIA GPUs, including strategy, 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

Qualifications

  • Experience with training/post training and optimization software (e.g., PyTorch distributed, torchtitan, VeRL, Nemo Framework, etc.)
  • Demonstrable knowledge of GenAI or ML concepts, especially model training, performance optimization, and software development/delivery
  • Experience with large scale distributed systems
  • BS or MS in Computer Science/Computer Engineering or equivalent experience
  • 15+ years of technical product management or similar experience at a technology company
  • Strong communication and interpersonal skills

Skills

Required:Training/post-training softwareGenAI/ML concepts

Benefits

  • Equity
  • Benefits (per NVIDIA benefits program)
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
Company profileWebsite

More open roles at NVIDIA