Senior Product Manager (SDK + Metrics & Storage)

San Francisco, California

Weights & Biases

Weights & Biases, developer tools for machine learning

View all jobs at Weights & Biases

Apply now Apply later

At Weights & Biases, our mission is to build the best tools for AI developers. We founded our company on the insight that while there were excellent tools for developers to build better code, there were no similarly great tools to help ML practitioners build better models. Starting with our first experiment tracking product, we have since expanded our solution into a comprehensive AI developer platform for organizations focused on building their own deep learning models and generative AI applications.
Weights & Biases is a Series C company with $250M in funding and over 200 employees. We proudly serve over 1,000 customers and more than 30 foundation model builders including customers such as OpenAI, NVIDIA, Microsoft, and Toyota.
Weights & Biases is looking for a Senior Product Manager to join our platform team leading the strategy and execution for our SDK and core backend systems powering experiment tracking—including storage and metrics infrastructure. This is a foundational role on our product team, supporting some of the most critical workflows in the MLOps lifecycle for thousands of ML practitioners around the world.

What You'll Do

  • The W&B SDK is the first touchpoint for most users and a core part of their daily workflow. It enables seamless logging of experiments, visualizations, model artifacts, and more—from any ML framework or environment. Behind the scenes, our storage and metrics systems handle massive volumes of data with speed, reliability, and scale to power those visualizations and experiments. You will set the product direction for the team, guiding the organization to balance short-term deliverables with strategic, long-term improvements.
  • Own and evolve the product strategy for the W&B SDK, ensuring it's intuitive, extensible, and loved by developers.
  • Partner closely with engineering to design and prioritize improvements to our metrics ingestion, storage, and retrieval systems.
  • Define and track success metrics for SDK adoption, reliability, and performance.
  • Work with developer advocates, GTM, and support teams to identify pain points, synthesize user feedback, and turn insights into product improvements.
  • Collaborate with security and infrastructure teams to ensure our systems scale across enterprise, SaaS, and airgapped deployments.
  • This role sits at the heart of W&B’s core value proposition: helping ML practitioners build faster, more reliable models with less friction.

Who You Are

  • Have 5+ years of product management experience, ideally with a background in developer tools, SDKs, or technical infrastructure.
  • Speak fluently with engineers about APIs, storage tradeoffs, performance bottlenecks, and distributed systems.
  • Understand the needs of ML engineers and researchers—or are excited to learn quickly.
  • Are passionate about building intuitive tools for highly technical users.
  • Take a structured approach to problem solving, but are comfortable navigating ambiguity and moving quickly.
  • Passionate about user research, integrating customer feedback, have deep curiosity and obsesses about the end to end developer experience.
  • Value clarity, empathy, communication and quality.

Bonus If You

  • Have experience working on Python libraries, open source tooling, or metrics systems.
  • Have worked on scaling or optimizing storage and ingestion pipelines at high throughput.
  • Have familiarity with MySQL, Kubernetes, Terraform, and major cloud providers (e.g., GCP, AWS, Azure).
  • Have dabbled in ML workflows and understand the unique needs of experimentation-heavy teams. If you have used W&B before, great!
We encourage you to apply even if your experience doesn't perfectly align with the job description as we seek out diverse and creative perspectives. Team members who love to learn and collaborate in an inclusive environment will flourish with us. We are an equal opportunity employer and do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status. If you need additional accommodations to feel comfortable during your interview process, reach out at careers@wandb.com.
#LI-Remote
Apply now Apply later

* Salary range is an estimate based on our AI, ML, Data Science Salary Index 💰

Job stats:  0  0  0

Tags: APIs AWS Azure Deep Learning Distributed Systems Engineering GCP Generative AI Kubernetes Machine Learning MLOps MySQL OpenAI Open Source Pipelines Python Research Security Terraform Weights & Biases

Regions: Remote/Anywhere North America
Country: United States

More jobs like this