Principal Product Manager, AI Frameworks
Core
Building tools, SDKs, and libraries for post-training RL to enable model builders to achieve large-scale performance and resilience on NVIDIA GPUs.
Role type
Principal Product Manager, AI Frameworks (Post-training & RL)
Builds
Post-training RL libraries, frameworks, SDKs, and training software for researchers and production model builders.
Domain
AI/ML, GPU computing, Large Language Models (GenAI), Distributed Systems
Deliverable
production ML models
Required skills
Product strategy & roadmap definition, Go-to-market planning, Large-scale distributed systems, GenAI/ML concepts, Training/post-training software design, Open Source ecosystem management, GPU architecture knowledge
Preferred skills
RL research-to-production leadership, Open Source/Github-first developer products, HW/SW co-design, Performance profiling
Technologies
VeRL, Tunix, Nemo Framework, PyTorch distributed, torchtitan
Responsibilities
Create and optimize post-training RL libraries and frameworks; 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, hands-on IC with strategic leadership
