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 large-scale model training and optimization
Domain
AI/ML infrastructure, GPU computing, distributed systems
Deliverable
product features
Required skills
Product strategy and roadmap definition, Go-to-market planning, GenAI and machine learning concepts, Large-scale distributed systems, Training/post-training software optimization
Preferred skills
GenAI/RecSys research to production leadership, Open Source/Github-first developer products, GPU architecture and HW/SW co-design
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
