Principal Product Manager, AI Frameworks
Core
Building tools, SDKs, and libraries to enable model builders to achieve large-scale performance and resilience on NVIDIA GPUs for training and post-training workloads.
Role type
Principal Product Manager, AI Frameworks
Builds
AI platform training tools, SDKs, and libraries for researchers and production model builders
Domain
AI/ML infrastructure, GPU computing, distributed systems
Deliverable
product features
Required skills
Product strategy and roadmap definition, Go-to-market planning, Large-scale distributed systems experience, GenAI and machine learning concepts, Training/post-training and optimization software expertise
Preferred skills
Leading GenAI/RecSys research to production at scale, Open Source and GitHub-first developer product experience, GPU architecture and HW/SW co-design knowledge
Technologies
PyTorch distributed, torchtitan, VeRL, Nemo Framework
Responsibilities
Create products for 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, Work with leadership to align with and drive company strategy
Seniority
Principal, strategy & mentorship
