A Flyte SDK (v2) version of this plugin is available as flyteplugins-mlflow.
MLflow
flytekitplugins-mlflow
MLflow enables us to log parameters, code, and results in machine learning experiments and compare them using an interactive UI.
pip install flytekitplugins-mlflowQuick Start(example, may need adjustment)
See full examplespip install flytekitplugins-mlflow
from flytekit import task, workflow
from flytekitplugins.mlflow import mlflow_autolog
@task
def my_task() -> None:
mlflow_autolog(...)
@workflow
def my_workflow() -> None:
my_task()Available Imports (1)
Task for MLflow.
from flytekitplugins.mlflow import mlflow_autolog
Dependencies
Related Plugins
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.
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.
Weights & Biases
The Weights and Biases MLOps platform helps AI developers streamline their ML workflow from end-to-end. This plugin
Weights & Biases
This plugin provides integration between Flyte and Weights & Biases (W&B) for experiment tracking, including support for distributed training with PyTorch Elastic.
