weights-and-biases

Track ML experiments, hyperparameter sweeps, and model artifacts with Weights & Biases.

Updated Aug 21, 2026
One-click install
npx skills add https://github.com/TylerSimons1127/vibe --skill weights-and-biases-tylersimons1127
Or copy as Structured Prompt for Agent▼
Please help me install this Agent Skill.
Skill: weights-and-biases
Source: https://github.com/TylerSimons1127/vibe/tree/main/skills/mlops/evaluation/weights-and-biases
Command: npx skills add https://github.com/TylerSimons1127/vibe --skill weights-and-biases-tylersimons1127

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires wandb, and includes references (resource) components.

What problem does it solve? Machine learning teams lose track of training runs, hyperparameters, and model versions when experiments are managed manually, making results impossible to compare or reproduce. ## Core Features & Use Cases - Experiment Tracking: Log metrics, configs, media, and system stats from PyTorch, TensorFlow, Keras, HuggingFace, and PyTorch Lightning training loops with wandb.init and wandb.log. - Hyperparameter Sweeps: Run grid, random, or Bayesian optimization searches with early termination and parallel agents across multiple GPUs. - Artifacts & Model Registry: Version datasets and models with lineage tracking, aliases, and promotion stages from staging to production. - Use Case: A team fine-tuning a ResNet50 classifier launches a 50-trial Bayesian sweep, compares validation accuracy across runs in a shared dashboard, and links the best checkpoint to the production model registry. ## Quick Start Ask the agent to instrument your PyTorch training script with W&B logging so every run's metrics and hyperparameters appear in a project dashboard.

Frequently Asked Questions about weights-and-biases

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I track PyTorch training experiments with Weights & Biases?▼

Call wandb.init with your project name and config dict, then call wandb.log inside your training loop with metrics like loss and accuracy. Finish with wandb.finish, and use wandb.save or Artifacts to upload model checkpoints.

How to run hyperparameter sweeps with wandb?▼

Define a sweep_config with a search method (grid, random, or bayes), a target metric, and parameter distributions, then create it with wandb.sweep. Launch trials with wandb.agent pointing to a train function that reads hyperparameters from wandb.config.

Does W&B integrate with HuggingFace Transformers and PyTorch Lightning?▼

Yes. Set report_to="wandb" in TrainingArguments for automatic HuggingFace logging, or pass WandbLogger to the PyTorch Lightning Trainer. Keras uses dedicated callbacks from wandb.integration.keras such as WandbMetricsLogger.

What is the difference between W&B Artifacts and the Model Registry?▼

Artifacts version any file-based asset like datasets or checkpoints with automatic lineage tracking between runs. The Model Registry is a curated layer where model artifacts are linked and promoted through aliases like staging and production.

Can I use wandb offline without an internet connection?▼

Yes. Set WANDB_MODE=offline before initializing, and all metrics are written locally. Sync them to the cloud later by running wandb sync on the run directory once connectivity is restored.

Which sweep search method should I choose in W&B?▼

Use grid search for a few discrete parameters needing full coverage, random search for quick exploration of many parameters, and Bayesian optimization when training runs are expensive and you need sample-efficient search. Bayesian is the recommended default.