Principal AI Platform & Frameworks PM (Training)

NVIDIA Corporation; Santa Clara (CA); On-site; USD 240,000 - 380,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA Corporation is seeking a Product Manager for AI Platform training to shape the tools, SDKs, and libraries that empower model builders to achieve top performance on NVIDIA GPUs.

You will partner with researchers and production teams to define roadmaps, go-to-market plans, and product-based roadmaps for training and post-training software, while aligning with company strategy and leadership.

Qualifications

• 15+ years of technical product management experience at a technology company.
• Experience with training/post-training and optimization software such as PyTorch distributed, torchtitan, VeRL, Nemo Framework.
• Experience with large-scale distributed systems.

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.

Skills

GenAI concepts

Distributed training software

Communication & interpersonal skills

Large-scale distributed systems

Product management experience

Education

BS/MS in CS/CE or related

Tools

PyTorch distributed

Nemo Framework

VeRL

torchtitan

Job description

NVIDIA Corporation is seeking a Product Manager for AI Platform training to shape the tools, SDKs, and libraries that empower model builders to achieve top performance on NVIDIA GPUs.

You will partner with researchers and production teams to define roadmaps, go-to-market plans, and product-based roadmaps for training and post-training software, while aligning with company strategy and leadership.