Staff Product Manager, Training Infrastructure - Weights & Biases

Role description

Role Overview

The Staff Product Manager will own the strategy and execution for products at the intersection of W&B Launch, infrastructure insights, and new product concepts. They will work directly with frontier model builders to understand their needs and ship products that accelerate deep learning workflows.

What You Will Do

Define and ship infrastructure-native products, own W&B Launch, and evolve it for the CoreWeave era. Spend real time with frontier AI teams, learn how they work, and build the bridge between W&B product and CoreWeave platform engineering.

Why It Might Be a Fit

The role requires experience as a product manager working on developer tools, deep learning infrastructure, or ML engineer platforms. The ideal candidate will have technical fluency, cross-functional influence, and a high bar for developer experience.

Requirements

• 7+ years as a product manager working on developer tools, deep learning infrastructure, or ML engineer platforms
• end-to-end ownership of technical products
• technical fluency in Kubernetes, SLURM, job scheduling, hyperparameter optimization algorithms, and API design
• cross-functional influence and ability to manage and influence AI engineers, software engineers, and stakeholders
• clear communication and ability to channel feedback into a single, executable stream

Benefits

• Medical, dental, and vision insurance
• Company-paid Life Insurance
• Voluntary supplemental life insurance
• Short and long-term disability insurance
• Flexible Spending Account
• Health Savings Account
• Tuition Reimbursement
• Ability to Participate in Employee Stock Purchase Program (ESPP)
• Mental Wellness Benefits through Spring Health
• Family-Forming support provided by Carrot
• Paid Parental Leave
• Flexible, full-service childcare support with Kinside
• 401(k) with a generous employer match
• Flexible PTO
• Catered lunch each day in our office and data center locations
• A casual work environment
• A work culture focused on innovative disruption