Wandb Python Packages

wandb-osh

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

substrax

JAX/Flax NNX training infrastructure: device meshes and SPMD sharding, an Orbax checkpoint store, early stopping and callbacks, W&B and MLflow logging.

calibrax

JAX benchmarking, profiling and evaluation metrics for Flax NNX: a registry of pure-function metrics (regression, classification, calibration, uncertainty, forecasting, generative, image, text, audio, graph, fairness), XLA FLOP counting, roofline analysis, GPU and energy monitoring, regression detection, publication exports, W&B and MLflow.

expr-tracker

Local-first experiment tracking with queryable history and expression-based alerts — catch NaN losses, divergence and stalled runs before you wake up

floppy-tracker

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

trloom

YAML-driven end-to-end fine-tuning on Hugging Face TRL — one config for model, dataset, W&B, and Modal GPUs.

torchexplorer

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

weightslab

Monitoring platform for ML teams building real‑world AI at scale.

ranzen

A python toolkit facilitating machine-learning experimentation.

rl-toolkit

RL-Toolkit: A Research Framework for Robotics

palkit

A python toolkit facilitating machine-learning experimentation.

mlxu

Machine Learning eXperiment Utilities

agentic-experiments

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

log-ui

A local, read-only, wandb-style dashboard for trackio experiment stores

llmtracer

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

built

BuilT - Build a Trainer of deep neural networks

fseval

Benchmarking framework for Feature Selection and Feature Ranking algorithms 🚀

video-transformers

Easiest way of fine-tuning HuggingFace video classification models

slp

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

wandb-allennlp

Utilities and boilerplate code to use wandb with allennlp

wandb-tui

Weights & Biases TUI

openvalidators

Repository for bittensor validators

kagglerecipes

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

slurm-to-wandb

Monitor SLURM jobs using Weights and Biases (wandb) 📊