# Product Manager for AI Platform Training

**Company**: NVIDIA
**Location**: Santa Clara, CA
**Experience**: senior
**Job type**: full-time
**Salary**: $240,000 - $379,500
**Category**: Product Management
**Industry**: Technology

**Apply**: https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/US-CA-Santa-Clara/Principal-Product-Manager--AI-Frameworks_JR2008001?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply
**Canonical**: https://yubhub.co/jobs/job_ffd5b99d-13c

## Description

As a Product Manager for AI Platform Training at NVIDIA, you will be responsible for building tools, SDKs, and libraries that enable model builders to achieve the best large-scale performance, resilience, and experience on NVIDIA GPUs.

The goal of NVIDIA Product Managers is to enable researchers and operators to succeed on the NVIDIA Platform and push the boundaries of what is possible in their research and research-to-production. You will work directly with customers inside and outside the company to identify key improvements, create roadmaps, and stay updated on the training and post-training landscape.

**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 and post-training software

- Work with leadership to align with and drive company strategy

**Requirements:**

- Experience with training and post-training optimization software (e.g., PyTorch distributed, Torchtitan, VeRL, Nemo Framework)

- 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

- 15+ years of technical product management experience at a technology company

- Strong communication and interpersonal skills

**Preferred Qualifications:**

- Experience leading GenAI/RecSys research to production at scale

- Working on Open Source and GitHub-first developer products with deep customer interactions

- Knowledge of GPU architecture, HW/SW co-design, and performance profiling

Your base salary will be determined based on your location, experience, and other objective job-related factors. The expected base salary range for this position is $240,000 - $379,500 USD.

You will also be eligible for equity and benefits.

## Skills

### Required
- PyTorch distributed
- Torchtitan
- VeRL
- Nemo Framework
- GenAI
- machine learning
- large-scale distributed systems
- Computer Science
- Computer Engineering

### Nice to have
- GenAI/RecSys research
- Open Source
- GitHub-first developer products
- GPU architecture
- HW/SW co-design
- performance profiling

---

Source: [Apply at nvidia.wd5.myworkdayjobs.com](https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/US-CA-Santa-Clara/Principal-Product-Manager--AI-Frameworks_JR2008001?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply)