Staff Product Manager, Training Infrastructure - Weights & Biases

CoreWeave, the AI Hyperscaler™, acquired Weights & Biases to create the most powerful end-to-end platform to develop, deploy, and iterate AI faster. Since 2017, CoreWeave has operated a growing footprint of data centers covering every region of the US and across Europe, and was ranked as one of the TIME100 most influential companies of 2024. By bringing together CoreWeave’s industry-leading cloud infrastructure with the best-in-class tools AI practitioners know and love from Weights & Biases, we’re setting a new standard for how AI is built, trained, and scaled. The integration of our teams and technologies is accelerating our shared mission: to empower developers with the tools and infrastructure they need to push the boundaries of what AI can do. From experiment tracking and model optimization to high-performance training clusters, agent building, and inference at scale, we’re combining forces to serve the full AI lifecycle — all in one seamless platform. Weights & Biases has long been trusted by over 1,500 organizations — including AstraZeneca, Canva, Cohere, OpenAI, Meta, Snowflake, Square,Toyota, and Wayve — to build better models, AI agents and applications. Now, as part of CoreWeave, that impact is amplified across a broader ecosystem of AI innovators, researchers, and enterprises. As we unite under one vision, we’re looking for bold thinkers and agile builders who are excited to shape the future of AI alongside us. If you're passionate about solving complex problems at the intersection of software, hardware, and AI, there's never been a more exciting time to join our team. About the Role Training a frontier model is as much an infrastructure problem as it is a modeling problem. When a run fails at hour 47 of a 72-hour job, the question is almost never "what's wrong with the code" — it's "what happened on the cluster." Node failures, network timeouts, storage bottlenecks, and scheduling decisions determine whether training runs succeed or fail, how much they cost, and how fast teams can iterate. Until now, the developer tools and the infrastructure telemetry have lived in completely separate worlds. The CoreWeave acquisition changes that. W&B can now build products that connect application-level experiment tracking with cluster health, hardware telemetry, and compute orchestration — a surface area that's hard to assemble without owning both the developer tools and the infrastructure. Reporting to the Director of Product Management, this Staff Product Manager will own the strategy and execution for the products at this intersection: W&B Launch (job submission and compute orchestration), infrastructure insights that leverage CoreWeave's unique telemetry and platform capabilities, and new product concepts — including agentic and automated research workflows — that only exist because W&B and CoreWeave are now one company. Sandboxes (managed development environments) are also part of this surface, but the core opportunity is building infrastructure-native developer tools that take advantage of the combined platform. You'll be working directly with frontier model builders — teams training on thousands of GPUs — to understand what they need and ship it. 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. What You'll Do Define and ship infrastructure-native products that only W&B + CoreWeave can build. This is the core of the role. The integration between W&B's experiment tracking and CoreWeave's infrastructure platform opens a new product surface — giving researchers visibility and control that spans their code, their runs, and the machines those runs execute on. You'll identify the highest-value opportunities in this design space, build conviction around them, and drive the roadmap from concept through launch and iteration. Own W&B Launch and e…