Integrations

W&B integrations make it fast and easy to set up experiment tracking and data versioning inside existing projects. Check out integrations for ML frameworks such as PyTorch, ML libraries such as Hugging Face, or cloud services such as Amazon SageMaker.

  • Examples: Try the code with notebook and script examples for each integration.
  • Video Tutorials: Learn to use W&B with YouTube video tutorials

Add wandb to any library

Azure OpenAI Fine-Tuning

How to Fine-Tune Azure OpenAI models using W&B.

Catalyst

How to integrate W&B for Catalyst, a Pytorch framework.

Cohere fine-tuning

How to Fine-Tune Cohere models using W&B.

Databricks

How to integrate W&B with Databricks.

DeepChecks

How to integrate W&B with DeepChecks.

DeepChem

How to integrate W&B with DeepChem library.

Docker

How to integrate W&B with Docker.

Farama Gymnasium

How to integrate W&B with Farama Gymnasium.

fastai

Hugging Face Transformers

Hugging Face Diffusers

Hugging Face AutoTrain

Hugging Face Accelerate

Training and inference at scale made simple, efficient and adaptable

Hydra

How to integrate W&B with Hydra.

Keras

Kubeflow Pipelines (kfp)

How to integrate W&B with Kubeflow Pipelines.

LightGBM

Track your trees with W&B.

Metaflow

How to integrate W&B with Metaflow.

MMEngine

MMF

How to integrate W&B with Meta AI’s MMF.

MosaicML Composer

State of the art algorithms to train your neural networks

OpenAI API

How to use W&B with the OpenAI API.

OpenAI Fine-Tuning

How to Fine-Tune OpenAI models using W&B.

OpenAI Gym

How to integrate W&B with OpenAI Gym.

PaddleDetection

How to integrate W&B with PaddleDetection.

PaddleOCR

How to integrate W&B with PaddleOCR.

Prodigy

How to integrate W&B with Prodigy.

PyTorch

PyTorch Geometric

Pytorch torchtune

PyTorch Ignite

How to integrate W&B with PyTorch Ignite.

PyTorch Lightning

Ray Tune

How to integrate W&B with Ray Tune.

SageMaker

How to integrate W&B with Amazon SageMaker.

Scikit-Learn

Simple Transformers

How to integrate W&B with the Transformers library by Hugging Face.

Skorch

How to integrate W&B with Skorch.

spaCy

Stable Baselines 3

How to integrate W&B with Stable Baseline 3.

TensorBoard

TensorFlow

W&B for Julia

How to integrate W&B with Julia.

XGBoost

Track your trees with W&B.

YOLOv5

Ultralytics

YOLOX

How to integrate W&B with YOLOX.


Last modified January 15, 2025: Update official product names (#989) (3788779)