Principal Product Manager, AI Frameworks

Create and optimize post-training RL libraries, frameworks 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

Experience with design and scaling of training/post training and optimization software (ex. VeRL, Tunix, Nemo Framework, PyTorch distributed, torchtitan, 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

Experience leading RL for 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

You will also be eligible for equity and benefits .

This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.