Neptune
flytekitplugins-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.
pip install flytekitplugins-neptuneQuick Start(example, may need adjustment)
See full examplespip install flytekitplugins-neptune
from flytekit import task, workflow
from flytekitplugins.neptune import neptune_scale_run, neptune_init_run
@task
def my_task() -> None:
neptune_scale_run(...)
@workflow
def my_workflow() -> None:
my_task()Available Imports (2)
Task for Neptune.
from flytekitplugins.neptune import neptune_scale_run
Task for Neptune.
from flytekitplugins.neptune import neptune_init_run
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.
MLflow
MLflow enables us to log parameters, code, and results in machine learning experiments and compare them using an interactive UI.
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.
