Staff Product Manager, Training Infrastructure - Weights & Biases

The role

This Staff Product Manager will own the strategy and execution for products at the intersection of W&B experiment tracking and CoreWeave infrastructure. You will define infrastructure-native products, evolve the W&B Launch orchestration product, and work with frontier AI teams to shape the roadmap.

What you will do

• Define and ship infrastructure-native products.
• Own and evolve W&B Launch for job submission and compute orchestration.
• Work with frontier AI teams to understand their workflows and needs.
• Partner with CoreWeave platform engineering and W&B engineering teams.
• Resolve friction points in job submission, compute management, and infrastructure observability.
• Explore new product concepts at the hardware-software boundary.

Required qualifications

• 7+ years as a product manager working on developer tools, deep learning infrastructure, or ML engineer platforms.
• Experience taking a technical product from definition to production.
• Comfortable discussing Kubernetes, SLURM, job scheduling, hyperparameter optimization algorithms, and API design.

Nice to have

• 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.

Benefits

• Medical, dental, and vision insurance - 100% paid for by CoreWeave
• 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

Before you apply

• Must be a U.S. person (citizen, lawful permanent resident, refugee, or asylee) or eligible to access export controlled information.

About the company

CoreWeave is a cloud provider for AI that acquired Weights & Biases to create an end-to-end platform for developing, deploying, and iterating AI.