FlyteKit Plugins

This is a list of all plugins that are available in FlyteKit.

Plugin Description
Apache Airflow This package holds the Airflow plugins for flytekit
Async FSSpec This package holds the data persistence plugins for flytekit
AWS Athena This package holds the Athena plugins for flytekit
AWS Batch This package holds the AWS Batch plugins for flytekit
AWS SageMaker Flytekit AWS SageMaker Plugin
Bigquery This package holds the Bigquery plugins for flytekit
Comet ML This package enables seamless use of Comet within Flyte
Dask Dask plugin for flytekit
FSSpec This is a deprecated plugin as of flytekit 1.5
DBT DBT Plugin for Flytekit
Deck This Plugin provides more renderers to improve task visibility
Dolt Dolt plugin for flytekit
DuckDB DuckDB Plugin for Flytekit
Envd This package enables users to easily build a Docker image for tasks or workflows.
Flyte Interactive This package holds the flyteinteractive plugins for flytekit
Great Expectations Great Expectations Plugin for Flytekit
Hive This package holds Hive plugins for flytekit
Hugging Face Hugging Face plugin for flytekit
Google IAP External command plugin to generate ID tokens for GCP Identity Aware Proxy
Inference This package enables seamless use of model inference sidecar services within Flyte
Kubernetes Pod Flytekit plugin to support K8s Pod tasks
Kubeflow MPI K8s based MPI plugin for flytekit
Kubeflow PyTorch K8s based Pytorch plugin for Flytekit
Kubeflow TensorFlow K8s based Tensorflow plugin for flytekit
Memray Profiling This package enables memory profiling for tasks with memray
MLflow This package enables seamless use of MLFlow within Flyte
Memory Machine Cloud MemVerge Flyte plugin
Modin Modin plugin for flytekit
Neptune This package enables seamless use of Neptune within Flyte
OmegaConf OmegaConf plugin for Flytekit
ONNX PyTorch ONNX PyTorch Plugin for Flytekit
ONNX ScikitLearn ONNX ScikitLearn Plugin for Flytekit
ONNX TensorFlow ONNX TensorFlow Plugin for Flytekit
OpenAI This package holds the openai plugins for flytekit
Optuna (wrapper) Optuna plugin for flytekit
Pandera Pandera plugin for flytekit
Papermill This is the flytekit papermill plugin
Perian Job Platform Flyte agent for Perian Job Platform (perian.io)
Polars Polars plugin for flytekit
Ray This package holds the Ray plugins for flytekit
Slurm This package holds the Slurm plugins for flytekit
Snowflake This package holds the Snowflake plugins for flytekit
Spark Spark 3 plugin for flytekit
SQLAlchemy SQLAlchemy plugin for flytekit
Vaex Vaex plugin for flytekit
Weights & Biases This package enables seamless use of Weights & Biases within Flyte
whylogs Enable the use of whylogs profiles to be used in flyte tasks to get aggregate statistics about data.