wandb0.23.1
Published
A CLI and library for interacting with the Weights & Biases API.
pip install wandb
Package Downloads
Authors
Project URLs
Requires Python
>=3.8
Dependencies
- click
>=8.0.1 - eval-type-backport
; python_version < "3.10" - gitpython
!=3.1.29,>=1.0.0 - packaging
- platformdirs
- protobuf
!=4.21.0,!=5.28.0,<7,>=3.12.0; python_version < "3.9" and sys_platform == "linux" - protobuf
!=4.21.0,!=5.28.0,<7,>=3.15.0; python_version == "3.9" and sys_platform == "linux" - protobuf
!=4.21.0,!=5.28.0,<7,>=3.19.0; python_version > "3.9" and sys_platform == "linux" - protobuf
!=4.21.0,!=5.28.0,<7,>=3.19.0; sys_platform != "linux" - pydantic
<3 - pyyaml
- requests
<3,>=2.0.0 - sentry-sdk
>=2.0.0 - typing-extensions
<5,>=4.8 - boto3
; extra == "aws" - botocore
>=1.5.76; extra == "aws" - azure-identity
; extra == "azure" - azure-storage-blob
; extra == "azure" - google-cloud-storage
; extra == "gcp" - filelock
; extra == "importers" - mlflow
; extra == "importers" - polars
<=1.2.1; extra == "importers" - rich
; extra == "importers" - tenacity
; extra == "importers" - google-cloud-storage
; extra == "kubeflow" - kubernetes
; extra == "kubeflow" - minio
; extra == "kubeflow" - sh
; extra == "kubeflow" - awscli
; extra == "launch" - azure-containerregistry
; extra == "launch" - azure-identity
; extra == "launch" - azure-storage-blob
; extra == "launch" - boto3
; extra == "launch" - botocore
>=1.5.76; extra == "launch" - chardet
; extra == "launch" - google-auth
; extra == "launch" - google-cloud-aiplatform
; extra == "launch" - google-cloud-artifact-registry
; extra == "launch" - google-cloud-compute
; extra == "launch" - google-cloud-storage
; extra == "launch" - iso8601
; extra == "launch" - jsonschema
; extra == "launch" - kubernetes
; extra == "launch" - kubernetes-asyncio
; extra == "launch" - nbconvert
; extra == "launch" - nbformat
; extra == "launch" - optuna
; extra == "launch" - pydantic
; extra == "launch" - pyyaml
>=6.0.0; extra == "launch" - tomli
; extra == "launch" - tornado
>=6.5.0; python_version >= "3.9" and extra == "launch" - typing-extensions
; extra == "launch" - bokeh
; extra == "media" - imageio
>=2.28.1; extra == "media" - moviepy
>=1.0.0; extra == "media" - numpy
; extra == "media" - pillow
; extra == "media" - plotly
>=5.18.0; extra == "media" - rdkit
; extra == "media" - soundfile
; extra == "media" - cloudpickle
; extra == "models" - orjson
; extra == "perf" - sweeps
>=0.2.0; extra == "sweeps" - wandb-workspaces
; extra == "workspaces"

Use W&B to build better models faster. Track and visualize all the pieces of your machine learning pipeline, from datasets to production machine learning models. Get started with W&B today, sign up for a W&B account!
Building an LLM app? Track, debug, evaluate, and monitor LLM apps with Weave, our new suite of tools for GenAI.
Documentation
See the W&B Developer Guide and API Reference Guide for a full technical description of the W&B platform.
Quickstart
Install W&B to track, visualize, and manage machine learning experiments of any size.
Install the wandb library
pip install wandb
Sign up and create an API key
Sign up for a W&B account. Optionally, use the wandb login CLI to configure an API key on your machine. You can skip this step -- W&B will prompt you for an API key the first time you use it.
Create a machine learning training experiment
In your Python script or notebook, initialize a W&B run with wandb.init().
Specify hyperparameters and log metrics and other information to W&B.
import wandb
# Project that the run is recorded to
project = "my-awesome-project"
# Dictionary with hyperparameters
config = {"epochs" : 1337, "lr" : 3e-4}
# The `with` syntax marks the run as finished upon exiting the `with` block,
# and it marks the run "failed" if there's an exception.
#
# In a notebook, it may be more convenient to write `run = wandb.init()`
# and manually call `run.finish()` instead of using a `with` block.
with wandb.init(project=project, config=config) as run:
# Training code here
# Log values to W&B with run.log()
run.log({"accuracy": 0.9, "loss": 0.1})
Visit wandb.ai/home to view recorded metrics such as accuracy and loss and how they changed during each training step. Each run object appears in the Runs column with generated names.
Integrations
W&B integrates with popular ML frameworks and libraries making it fast and easy to set up experiment tracking and data versioning inside existing projects.
For developers adding W&B to a new framework, follow the W&B Developer Guide.
W&B Hosting Options
Weights & Biases is available in the cloud or installed on your private infrastructure. Set up a W&B Server in a production environment in one of three ways:
- Multi-tenant Cloud: Fully managed platform deployed in W&B’s Google Cloud Platform (GCP) account in GCP’s North America regions.
- Dedicated Cloud: Single-tenant, fully managed platform deployed in W&B’s AWS, GCP, or Azure cloud accounts. Each Dedicated Cloud instance has its own isolated network, compute and storage from other W&B Dedicated Cloud instances.
- Self-Managed: Deploy W&B Server on your AWS, GCP, or Azure cloud account or within your on-premises infrastructure.
See the Hosting documentation in the W&B Developer Guide for more information.
Python Version Support
We are committed to supporting our minimum required Python version for at least six months after its official end-of-life (EOL) date, as defined by the Python Software Foundation. You can find a list of Python EOL dates here.
When we discontinue support for a Python version, we will increment the library’s minor version number to reflect this change.
Contribution guidelines
Weights & Biases ❤️ open source, and we welcome contributions from the community! See the Contribution guide for more information on the development workflow and the internals of the wandb library. For wandb bugs and feature requests, visit GitHub Issues or contact [email protected].
W&B Community
Be a part of the growing W&B Community and interact with the W&B team in our Discord. Stay connected with the latest ML updates and tutorials with W&B Fully Connected.