Principal Product Manager, AI Frameworks
Core
Building tools, SDKs, and libraries for post-training RL to enable model builders to achieve high performance and resilience on NVIDIA GPUs.
Role type
Principal Product Manager, AI Frameworks (Post-training & RL)
Builds
Post-training RL libraries, frameworks, SDKs, and optimization software for large-scale model training.
Domain
AI/ML Infrastructure, GPU Computing, Distributed Systems
Deliverable
production ML models | product features
Required skills
Product strategy & roadmap definition, Go-to-market planning, Large-scale distributed systems design, GenAI/ML model training concepts, Software development lifecycle, Open source ecosystem management, GPU architecture knowledge, Performance profiling, Stakeholder alignment with leadership.
Preferred skills
Leading RL research-to-production at scale, Deep Open Source/GitHub-first developer product experience, HW/SW co-design expertise.
Technologies
VeRL, Tunix, Nemo Framework, PyTorch distributed, torchtitan, OSS frameworks.
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 for training/post-training software; Work with leadership to align with and drive company strategy.
Seniority
Principal, strategy & mentorship
