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.

ranzen

A python toolkit facilitating machine-learning experimentation.

agentic-experiments

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

torchexplorer

Interactively inspect pytorch modules during training.

rl-toolkit

RL-Toolkit: A Research Framework for Robotics

palkit

A python toolkit facilitating machine-learning experimentation.

video-transformers

Easiest way of fine-tuning HuggingFace video classification models

wandb-allennlp

Utilities to use allennlp with wandb

built

BuilT - Build a Trainer of deep neural networks

floppy-tracker

A hardware-agnostic python library to monitor the computational cost 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

slp

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

fseval

Benchmarking framework for Feature Selection and Feature Ranking algorithms 🚀

kagglerecipes

Useful code for different Kaggle competitions. Currently covering RSNA-MICCAI Brain Tumor Radiogenomic Classification Kaggle competition.

mlxu

Machine Learning eXperiment Utilities

openvalidators

Repository for bittensor validators

umarise-wandb

Anchor W&B artifacts to Bitcoin. Zero-touch provenance for ML experiments.

wandbfsspec

🍱 fsspec interface for Weights & Biases (wandb)

slurm-to-wandb

Monitor SLURM jobs using Weights and Biases (wandb) 📊

openvalidators-test

Repository for bittensor validators

wandbot

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

easysweeps

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

wandb-preempt

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