What problem does it solve?
Weights & Biases (W&B) eliminates scattered experiment tracking by centralizing metrics, configurations, artifacts, and model registry workflows so teams can compare results and reproduce outcomes.
Core Features & Use Cases
- Experiment tracking (projects & runs): record configs, metrics, run IDs, and shareable run URLs for fast comparison across experiments.
- Real-time visualization (metrics & custom plots): log scalars, images, tables, histograms, and charts to understand training dynamics as they happen.
- Hyperparameter optimization (sweeps): run automated sweeps (grid/random/bayesian) to find better learning rates and training configurations efficiently.
- Artifacts & model registry: version datasets and models with lineage, aliases (latest/best/production), and staged promotion for collaboration and deployment.
- Framework integrations: work with popular training stacks such as Hugging Face Transformers, PyTorch Lightning, Keras/TensorFlow, and PyTorch native loops.
Quick Start
Install W&B with pip, log in with your API key using wandb login, then initialize a run via wandb.init(project="your-project") and log metrics with wandb.log during your training loop.