Staff Product Manager, Training Infrastructure - Weights & Biases

About the role

The Staff Product Manager will own the strategy and execution for the products at the intersection of W&B and CoreWeave. This includes defining and shipping infrastructure-native products, evolving W&B Launch, and exploring new product concepts.

Responsibilities

• Define and ship infrastructure-native products that only W&B + CoreWeave can build.
• Own W&B Launch and evolve it for the CoreWeave era.
• Spend real time with frontier AI teams and learn how they actually work.
• Build the bridge between W&B product and CoreWeave platform engineering.
• Find and resolve friction points in the job submission, compute management, and infrastructure observability workflows.
• Explore new product concepts at the hardware-software boundary.

Requirements

• Experience: 7+ years as a product manager working on developer tools, deep learning infrastructure, or ML engineer platforms.
• End-to-end ownership: Take a technical product from definition to production, including defining success criteria, measuring effectiveness, and iterating to improve impact and user happiness.
• Technical fluency: Comfortable discussing Kubernetes and SLURM, job scheduling, hyperparameter optimization algorithms, and API design with engineers and AI researchers.
• Cross-functional influence: Manage and influence AI engineers, software engineers, and stakeholders without official authority to align teams and drive outcomes.
• Clear communication: Clarify complex, fast-moving environments and channel feedback into executable streams.
• Care and taste: Have strong opinions about the products you use and a high bar for developer experience.

Qualifications

• Experience with deep learning frameworks (PyTorch, JAX), distributed training, or LLM development workflows.
• Hands-on experience with SLURM or Kubernetes for GPU-intensive workloads.
• Prior exposure to W&B or other experiment-tracking and model management platforms.
• Understanding of hyperparameter optimization methods and evaluation workflows for large models.

Skills

• Strong communication and collaboration skills.
• Ability to work independently and take initiative.
• Passion for improving developer experience and accelerating AI workflows.

Benefits

• Medical, dental, and vision insurance.
• Company-paid Life Insurance.
• Voluntary supplemental life insurance.
• Short and long-term disability insurance.
• Flexible Spending Account (FSA).
• Health Savings Account (HSA).
• 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.
• Catered lunch each day in our office and data center locations.
• A casual work environment.
• A work culture focused on innovative disruption.