A Flyte SDK (v2) version of this plugin is available as flyteplugins-wandb.
Weights & Biases
flytekitplugins-wandb
The Weights and Biases MLOps platform helps AI developers streamline their ML workflow from end-to-end. This plugin
pip install flytekitplugins-wandbQuick Start(example, may need adjustment)
See full examplespip install flytekitplugins-wandb
from flytekit import task, workflow
from flytekitplugins.wandb import wandb_init
@task
def my_task() -> None:
wandb_init(...)
@workflow
def my_workflow() -> None:
my_task()Available Imports (1)
Task for Weights & Biases.
from flytekitplugins.wandb import wandb_init
Dependencies
Related Plugins
Weights & Biases
This plugin provides integration between Flyte and Weights & Biases (W&B) for experiment tracking, including support for distributed training with PyTorch Elastic.
Comet ML
Comet’s machine learning platform integrates with your existing infrastructure and tools so you can manage, visualize, and optimize models—from training runs to production monitoring. This plugin integrates Flyte with Comet.ml by configuring links between the two platforms.
MLflow
MLflow enables us to log parameters, code, and results in machine learning experiments and compare them using an interactive UI.
Neptune
Neptune is the MLOps stack component for experiment tracking. It offers a single place to log, compare, store, and collaborate on experiments and models. This plugin integrates Flyte with Neptune by configuring links between the two platforms.
