Wandb Python Packages

wandb-osh

A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!

weightslab

PyTorch dataset debugger for computer vision — pause training, mine live loss signals to surface mislabels, class imbalance & outliers, then curate your image, video & LiDAR data without restarting

video-transformers

Easiest way of fine-tuning HuggingFace video classification models

calibrax

Unified benchmarking and profiling framework for the JAX scientific ML ecosystem. Timing, GPU/energy monitoring, FLOPS counting, roofline analysis, statistical testing, regression detection, and CI integration.

wandb-allennlp

Utilities and boilerplate code to use wandb with allennlp

kagglerecipes

Helpful data preprocessing, training, and visualisation code and scripts for a range of Kaggle competitions, supported by Weights & Biases.

ranzen

A python toolkit facilitating machine-learning experimentation.

torchexplorer

Interactively inspect module inputs, outputs, parameters, and gradients.

agentic-experiments

Hypothesis-first experiment tracking for agent-driven ML research, wired into Claude Code.

llmtracer

Trace LLM calls (and others) and visualize them in WandB, as interactive SVG or using a streaming local webapp

mlxu

Machine learning experiment utils.

palkit

A python toolkit facilitating machine-learning experimentation.

built

BuilT - Build a Trainer of deep neural networks

rl-toolkit

RL-Toolkit: A Research Framework for Robotics

fseval

Benchmarking framework for Feature Selection and Feature Ranking algorithms 🚀

floppy-tracker

A hardware-agnostic profiler for tracking the FLOPs and BOPs of Machine and Deep Learning algorithms.

slp

Utils and modules for Speech Language and Multimodal processing using pytorch and pytorch lightning

easysweeps

Easy Sweeps: A command-line utility for automating W&B sweep creation, launch, and GPU process management with ease.

umarise-wandb

Anchor every W&B artifact to Bitcoin. Async, with reconciliation ledger.

openvalidators

Openvalidators is a collection of open source validators for the Bittensor Network.

wandbfsspec

🍱 fsspec interface for Weights & Biases (wandb)

openvalidators-test

Openvalidators is a collection of open source validators for the Bittensor Network.

slurm-to-wandb

Uploads SLURM job information to WandB for better monitoring.

wandbot

wandbot is a technical support bot for Weights & Biases' AI developer tools that can run in Discord, Slack, ChatGPT and Zendesk