Wandb Python Packages

wandb-osh

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

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.

agentic-experiments

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

video-transformers

Easiest way of fine-tuning HuggingFace video classification models

torchexplorer

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

wandb-allennlp

Utilities and boilerplate code to use wandb with allennlp

ranzen

A python toolkit facilitating machine-learning experimentation.

rl-toolkit

RL-Toolkit: A Research Framework for Robotics

built

BuilT - Build a Trainer of deep neural networks

floppy-tracker

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

llmtracer

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

kagglerecipes

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

slp

Speech, Language and Multimodal Processing models and utilities in PyTorch

palkit

Useful functions.

fseval

Benchmarking framework for Feature Selection and Feature Ranking algorithms 🚀

mlxu

Machine Learning eXperiment Utilities

openvalidators

Repository for bittensor validators

umarise-wandb

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

slurm-to-wandb

Monitor SLURM jobs using Weights and Biases (wandb) 📊

openvalidators-test

Repository for bittensor validators

wandbfsspec

🍱 fsspec interface for Weights & Biases (wandb)

easysweeps

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

wandbot

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

wandb-preempt

Code and tutorial on integrating wandb sweeps with Slurm pre-emption