Staff Product Manager, Training Infrastructure - Weights & Biases

Weights & Biases; Sunnyvale (CA); Hybrid; USD 188,000 - 275,000; Full time

Benefits offered by this job

Medical, dental, and vision insurance - 100% paid

Tuition Reimbursement

401(k) with generous employer match

Flexible PTO

Casual work environment

Job summary

Weights & Biases in Sunnyvale is seeking a Staff Product Manager to lead the integration of developer tools and infrastructure following the CoreWeave acquisition. You will define and ship infrastructure-native products, owning the vision for W&B Launch. The ideal candidate has over 7 years of product management experience, a strong technical background in deep learning infrastructure, and exceptional cross-functional influence skills. We offer a hybrid work model along with competitive compensation.

Qualifications

• 7+ years as a product manager working on developer tools or ML engineer platforms.
• End-to-end ownership of technical products from concept to production.
• Ability to discuss job scheduling and hyperparameter optimization with engineers.

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 to learn their workflows.
• Partner with infrastructure and platform teams to identify integration opportunities.

Skills

Product management

Deep learning infrastructure

Technical fluency

Communication skills

Cross-functional influence

Tools

Kubernetes

SLURM

Job description

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. 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, this 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 evolve it for the CoreWeave era. Launch is W&B’s job submission and compute orchestration product. Today it helps teams submit training jobs to Kubernetes clusters and cloud providers. As part of a vertically integrated AI platform, Launch can go further — with deeper infrastructure awareness and the ability to serve as the orchestration layer for agentic and automated research workflows. You’ll set the vision for what that looks like and ship it.
• Spend real time with frontier AI teams and learn how they actually work. Foundation model builders are your primary customers. You’ll spend significant time in their workflows — understanding how they manage infrastructure, debug failures, orchestrate computation, and increasingly automate their research processes. These conversations will directly shape the product roadmap.
• Build the bridge between W&B product and CoreWeave platform engineering. This role sits at the seam between two organizations that are becoming one. You’ll partner with CoreWeave’s infrastructure and platform teams to identify integration opportunities, and with W&B’s engineering teams to build them into products researchers actually want to use. Your ability to translate between infrastructure engineers and ML researchers will be critical.
• Find and resolve friction points in the job submission, compute management, and infrastructure observability workflows, maintaining W&B’s high bar for craft and quality.
• Explore new product concepts at the hardware-software boundary. The design space opened by the CoreWeave integration is large — spanning profiling, environments, automation, agents, and more. You’ll evaluate which new concepts have the highest impact for frontier AI teams and drive the most promising ones from zero to one.

Who You Are

• Experience: You have 7+ years as a product manager working on developer tools, deep learning infrastructure, or ML engineer platforms.
• 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.
• Technical fluency: You’re comfortable discussing Kubernetes and SLURM, job scheduling, hyperparameter optimization algorithms, and API design with engineers and AI researchers.
• 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.
• 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.
• Care and taste: You have strong opinions about the products you use and a high bar for developer experience.

Preferred

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

Wondering if You’re a Good Fit?

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. Here Are a Few Qualities We’ve Found Compatible With Our Team

• Outgoing and user-focused. 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
• Self-directed and willing to take risks. At our speed and scale, product managers need to proactively find solutions, improve processes, and collaborate with team members and engaged users. Your initiative will really shine here.
• Hands-on with AI. You experiment with generative AI to improve your own workflow and build user-facing prototypes. You keep up to date every day with what’s new in the AI / LLM world.
• Passionate about developer experience. 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.

Why Us?

We work hard, have fun, and move fast! We’re in an exciting stage of hyper-growth that you will not want to miss out on. We’re not afraid of a little chaos, and we’re constantly learning. Our team cares deeply about how we build our product and how we work together, which is represented through our core values:

• Be Curious at Your Core
• Act Like an Owner
• Empower Employees
• Deliver Best-in-Class Client Experiences
• Achieve More Together

We support and encourage an entrepreneurial outlook and independent thinking. We foster an environment that encourages collaboration and provides the opportunity to develop innovative solutions to complex problems. As we get set for takeoff, the growth opportunities within the organization are constantly expanding. You will be surrounded by some of the best talent in the industry, who will want to learn from you, too. Come join us!

The base salary range for this role is $188,000 to $275,000. The starting salary will be determined based on job-related knowledge, skills, experience, and market location. We strive for both market alignment and internal equity when determining compensation. In addition to base salary, our total rewards package includes a discretionary bonus, equity awards, and a comprehensive benefits program (all based on eligibility).

What We Offer

The range we’ve posted represents the typical compensation range for this role. To determine actual compensation, we review the market rate for each candidate which can include a variety of factors. These include qualifications, experience, interview performance, and location.

In addition to a competitive salary, we offer a variety of benefits to support your needs, including:

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

Our Workplace

While we prioritize a hybrid work environment, remote work may be considered for candidates located more than 30 miles from an office, based on role requirements for specialized skill sets. New hires will be invited to attend onboarding at one of our hubs within their first month. Teams also gather quarterly to support collaboration

California Consumer Privacy Act - California applicants only

CoreWeave is an equal opportunity employer, committed to fostering an inclusive and supportive workplace. All qualified applicants and candidates will receive consideration for employment without regard to race, color, religion, sex, disability, age, sexual orientation, gender identity, national origin, veteran status, or genetic information.

As part of this commitment and consistent with the Americans with Disabilities Act (ADA), CoreWeave will ensure that qualified applicants and candidates with disabilities are provided reasonable accommodations for the hiring process, unless such accommodation would cause an undue hardship. If reasonable accommodation is needed, please contact: [e-mail].

Export Control Compliance

This position requires access to export controlled information. To conform to U.S. Government export regulations applicable to that information, applicant must either be (A) a U.S. person, defined as a U.S. citizen or national, (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. CoreWeave may, for legitimate business reasons, decline to pursue any export licensing process.