Principal Product Manager, AI Frameworks

Description

The company builds open source AI frameworks and supports a wide range of other open source frameworks, helping customers at every stage of their AI work on its GPUs.

This role covers the tools, SDKs and libraries that help model builders get strong performance, resilience and ease of use at large scale on the company's GPUs.

You would join a small product management group that serves frontier model builders working on training and post training, across all deep learning GPU use cases.

The work centres on the training and post training landscape, which you will track to keep product direction current.

The role suits someone who combines product skills, technical depth and enthusiasm for the work.

Responsibilities

• Build products that help researchers and teams training models in production.
• Set product strategy and write roadmaps and go-to-market plans.
• Work with internal and external customers to shape roadmaps for training and post training software.
• Identify key improvements through direct work with customers.
• Partner with company leadership to align product work with company strategy and help drive it.

Requirements

• Hands-on experience with training, post training or optimization software such as PyTorch distributed, torchtitan, VeRL or NeMo Framework.
• Working knowledge of GenAI or machine learning, particularly model training, performance optimization, and building and delivering software.
• Experience with large-scale distributed systems.
• A BS or MS in computer science, computer engineering or a similar field, or equivalent experience.
• 15+ years of technical product management or similar experience at a technology company.
• Strong communication and interpersonal skills.
• Ways to Stand Out: experience taking GenAI or recommender systems research to production at scale.
• Ways to Stand Out: experience with open source and GitHub-first developer products that involve close customer interaction.
• Ways to Stand Out: knowledge of GPU architecture, hardware and software co-design, and performance profiling.

Conditions

Base salary ranges from 240,000 USD to 379,500 USD, set by location, experience and other objective job-related factors.

Applications are accepted at least until November 3, 2026.

This posting is for an existing vacancy.

The company uses AI tools in its recruiting process.

The company is an equal opportunity employer and does not discriminate on any basis protected by law.