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.

Popular integrations

The following lists popular integrations used by W&B users:

ML frameworks

• Keras
• PyTorch
• PyTorch Lightning
• PyTorch Ignite
• TensorFlow
• Fastai
• Scikit-Learn

ML libraries

• Hugging Face
• PyTorch Geometric
• spaCy
• Ultralytics
• YOLOv5
• Simple Transformers
• Catalyst
• XGBoost
• LightGBM

Tools

• TensorBoard
• Amazon SageMaker
• Kubeflow Pipelines
• Dagster
• Docker
• Databricks
• Ray Tune
• OpenAI Gym

Related resources

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