Staff Product Manager (Training Infrastructure, Weights & Biases)

Join Weights & Biases, a leading platform for machine learning practitioners. As a Staff Product Manager, you will own the strategy and execution for products at the intersection of W&B Launch, infrastructure insights, and new product concepts. You will work directly with frontier model builders, define and ship infrastructure-native products, and build the bridge between W&B product and CoreWeave platform engineering. This role requires technical fluency, end-to-end ownership, and a passion for building tools that help AI practitioners accelerate their iteration and discovery.

Définir et expédier des produits natifs à l'infrastructure qui ne peuvent être construits que par W&B et CoreWeave, en identifiant les opportunités de valeur les plus élevées dans cet espace de conception.

End-to-end ownership: You've taken a technical product—whether a model, orchestration feature, or developer tool—all the way to production, including defining success criteria, measuring effectiveness, and iterating to improve impact and user happiness

About CoreWeave

Posséder W&B Launch et l'évoluer pour l'ère CoreWeave, en définissant la vision de ce à quoi cela ressemble et en l'expédiant.

If you're energized by solving infrastructure challenges that directly accelerate deep learning workflows, collaborating with the world’s best AI teams, and delivering tools that move the industry frontier forward, his role is for you

We believe in investing in our people and value candidates who can bring their own diversified experiences to our teams—even if you aren't a 100% skill or experience match

Care and taste: You have strong opinions about the products you use and a high bar for developer experience

Responsibilities

• Passer du temps réel avec des équipes d'IA de pointe pour comprendre leurs besoins et façonner la feuille de route du produit en conséquence.
• Technical fluency: You’re comfortable discussing Kubernetes and SLURM, job scheduling, hyperparameter optimization algorithms, and API design with engineers and AI researchers
• You care deeply about building tools that help AI practitioners accelerate their iteration and discovery. You know small friction points compound, and smoothing them out is what makes a delightful and winning developer tool
• Cross-functional influence: You’re skilled at managing and influencing AI engineers, software engineers, and stakeholders—with or without official authority—to align teams and drive outcomes

Requirements

• Clear communication: A major part of this role is seeking feedback from many directions and channeling it into a single, executable stream. You bring clarity and organization to complex, fast-moving environments
• Experience: You have 7+ years as a product manager working on developer tools, deep learning infrastructure, or ML engineer platforms
• You'll love this role if you enjoy connecting with real users day to day, helping them solve issues, and sharing patterns for making the most of our products
• (B) eligible to access the export controlled information without a required export authorization, or (C) eligible and reasonably likely to obtain the required export authorization from the applicable U.S. government agency
• Must either be (A) a U.S. person, defined as a (i) U.S. citizen or national, (ii) U.S. lawful permanent resident (green card holder), (iii) refugee under 8 U.S.C. § 1157, or (iv) asylee under 8 U.S.C. § 1158
• Understanding of hyperparameter optimization methods and evaluation workflows for large models

Preferred

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

Pay

USD 188,000 - 275,000