Principal Product Manager, AI Frameworks
Core
Building tools, SDKs, and libraries to enable model builders to achieve best-in-class large-scale performance, resilience, and experience on NVIDIA GPUs for training and post-training workloads.
Role type
Principal Product Manager, AI Frameworks
Builds
Training and post-training software frameworks, SDKs, and libraries for large-scale distributed systems.
Domain
AI/ML infrastructure, GPU computing, distributed systems
Deliverable
production ML models
Required skills
Product strategy and roadmap definition, GenAI and machine learning concepts, large-scale distributed systems, training/post-training optimization software (PyTorch, torchtitan, VeRL, Nemo), GPU architecture and HW/SW co-design, performance profiling
Preferred skills
Leading GenAI/RecSys research to production at scale, Open Source and GitHub-first developer products
Technologies
PyTorch, torchtitan, VeRL, Nemo Framework, NVIDIA GPUs
Responsibilities
Create products for researchers and production model builders, develop product strategy and go-to-market plans, collaborate with internal and external customers to build roadmaps, align with and drive company strategy
Seniority
Principal, hands-on IC with strategic leadership
