Senior Product Manager, AI Frameworks

What You'll Be Doing

• Create and optimize pre-training/inference and post-training frameworks 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 for the E2E ML lifecycle
• Work with leadership to align with and drive company strategy

What We Need To See

• Experience with design and scaling of training/inference post training and optimization software (Torchtitan, FSDP)
• Demonstrable knowledge of GenAI or machine learning concepts, particularly around model training, performance optimization, inference, and software development and delivery
• Experience with large scale distributed systems
• BS or MS degree in Computer Science, Computer Engineering, or similar experience (or equivalent experience)
• 10+ years of technical product management, or similar, experience at a technology company
• Strong communication and interpersonal skills

How To Stand Out

• Background with leading OSS RL Frameworks like VeRL
• Experience with orchestration and reliability at scale
• Experience working on Open Source & Github-first developer products with deep customer interactions
• Knowledge of GPU architecture, HW/SW co-design, and performance profiling

Base Salary Range

$168,000 - $258,750 for Level 4, and $208,000 - $327,750 for Level 5.

Application Information

Applications for this job will be accepted at least until July 7, 2026. This posting is for an existing vacancy.