Partners & Integrations

One platform.
Every tool you already use.

Union.ai runs your existing stack on one durable runtime: agent SDKs, models, data platforms, ML frameworks, & serving. Every integration below is tested, supported, & shipped by Union.ai.

Infrastructure

Deploy Union.ai anywhere

Run Union.ai on the cloud, region, and hardware that fits your workload.

AWS

Docs ↗

Managed data plane in your AWS account, or self-managed on EKS.

Azure

Docs ↗

Managed data plane on Azure, or self-managed on AKS.

CoreWeave

Docs ↗

GPU-dense deployments on CoreWeave's Kubernetes cloud.

Crusoe

Docs ↗

Run on Crusoe's energy-efficient GPU cloud.

GCP

Docs ↗

Managed data plane on GCP, or self-managed on GKE.

Nebius

Docs ↗

Deploy on Nebius AI cloud infrastructure.

On-prem & any Kubernetes

Docs ↗

Self-managed deployment on any conformant Kubernetes cluster.

Oracle Cloud

Docs ↗

Deploy on Oracle Cloud Infrastructure.

Storage

Your data stays in your account

Workflow data lives in your own object store, with native support for every major cloud.

AWS S3

Docs ↗

Native S3 support for workflow inputs, outputs, and artifacts.

Azure Blob Storage

Docs ↗

Native Azure Blob support for workflow data.

Google Cloud Storage

Docs ↗

Native GCS support for workflow data.

Agent Frameworks

Durable by default

Keep the agent SDK you already write in. Union.ai wraps each framework's own idioms with a durable runtime: crash-safe replay, retries, per-tool containers, cross-run memory, and full observability. Start with the agent frameworks overview.

Claude Agent SDK

Docs ↗

Give Claude-powered agents durable execution, retries, and replay across restarts.

pip install flyteplugins-agents-claude
pip install flyteplugins-agents-claude

CrewAI

Docs ↗

Run CrewAI crews with durable tool calls and automatic recovery.

pip install flyteplugins-agents-crewai
pip install flyteplugins-agents-crewai

Deep Agents

Docs ↗

LangChain Deep Agents with checkpointed planning and sub-agent steps.

pip install flyteplugins-agents-deepagents
pip install flyteplugins-agents-deepagents

Google ADK

Docs ↗

Run Google Agent Development Kit agents on Union.ai's durable runtime.

pip install flyteplugins-agents-google
pip install flyteplugins-agents-google

Hermes

Docs ↗

Union.ai's native agent framework, durable by design.

pip install flyteplugins-agents-hermes
pip install flyteplugins-agents-hermes

LangChain

Docs ↗

Run LangChain chains and agents with durable, resumable execution.

pip install flyteplugins-agents-langchain
pip install flyteplugins-agents-langchain

LangGraph

Docs ↗

Persist LangGraph state graphs across failures without changing your graph code.

pip install flyteplugins-agents-langgraph
pip install flyteplugins-agents-langgraph

MCP

Docs ↗

Serve Model Context Protocol servers as managed, multi-tenant Union.ai apps.

Built In

Mistral

Docs ↗

Durable agents on the Mistral Conversations API.

pip install flyteplugins-agents-mistral
pip install flyteplugins-agents-mistral

OpenAI Agents SDK

Docs ↗

Run OpenAI Agents SDK agents as durable workflows with checkpointed tool calls.

pip install flyteplugins-agents-openai
pip install flyteplugins-agents-openai

Pydantic AI

Docs ↗

Type-safe Pydantic AI agents running on a durable backend.

pip install flyteplugins-agents-pydantic-ai
pip install flyteplugins-agents-pydantic-ai
Models & Inference

Frontier APIs to self-hosted

Serve open models on Union.ai-managed GPUs with the inference engine of your choice.

Ollama

Docs ↗

Run Ollama model servers as Union.ai apps.

Built In

SGLang

Docs ↗

High-throughput LLM serving with SGLang, deployed as a managed app.

pip install flyteplugins-sglang
pip install flyteplugins-sglang

vLLM

Docs ↗

Serve open-weight LLMs with vLLM as an autoscaling Union.ai app.

pip install flyteplugins-vllm
pip install flyteplugins-vllm
Apps

UIs and APIs without glue code

Deploy common app types next to your workflows without writing the integration glue yourself. Gradio, Django, Panel, and WebSocket apps work out of the box too — see the apps guide.

Connector apps

Docs ↗

Deploy long-running connectors to external systems as managed apps.

Built In

FastAPI

Docs ↗

Serve REST APIs as autoscaling Union.ai apps with native FastAPI support.

Built In

Streamlit

Docs ↗

Deploy Streamlit dashboards alongside the pipelines that feed them.

Built In

Webhooks

Docs ↗

Trigger workflows from external events with managed webhook endpoints.

Built In
Data & Analytics

Process data where it lives

Push compute to your warehouse or lakehouse, spin up ephemeral clusters on demand, and move typed DataFrames between tasks.

BigQuery

Docs ↗

Run BigQuery SQL as workflow tasks with typed inputs and outputs.

pip install flyteplugins-bigquery
pip install flyteplugins-bigquery

Dask

Docs ↗

On-demand Dask clusters for distributed Python, managed per workflow.

pip install flyteplugins-dask
pip install flyteplugins-dask

Databricks

Docs ↗

Submit Spark jobs to Databricks from your workflows.

pip install flyteplugins-databricks
pip install flyteplugins-databricks

Hugging Face Datasets

Docs ↗

Move HF Datasets between tasks and reference Hub datasets directly.

pip install flyteplugins-huggingface
pip install flyteplugins-huggingface

JSONL

Docs ↗

Fast JSONL file and directory types with optional compression and Arrow.

pip install flyteplugins-jsonl
pip install flyteplugins-jsonl

Lance

Docs ↗

Lance columnar format for DataFrames, streamed lazily between tasks.

pip install flyteplugins-lance
pip install flyteplugins-lance

Pandera

Docs ↗

Schema-validated DataFrames (pandas, Polars, or PySpark) as task types.

pip install flyteplugins-pandera
pip install flyteplugins-pandera

Polars

Docs ↗

Pass Polars DataFrames and LazyFrames between tasks, backed by Parquet.

pip install flyteplugins-polars
pip install flyteplugins-polars

Snowflake

Docs ↗

Parameterized Snowflake queries, key-pair auth, and DataFrame results.

pip install flyteplugins-snowflake
pip install flyteplugins-snowflake

Spark

Docs ↗

Ephemeral Spark clusters on Kubernetes, sized per task and torn down after.

pip install flyteplugins-spark
pip install flyteplugins-spark
ML Frameworks

Train at any scale

Distributed training and large-scale compute with the frameworks your team already uses.

PyTorch

Docs ↗

Distributed and elastic PyTorch training with automatic fault recovery.

pip install flyteplugins-pytorch
pip install flyteplugins-pytorch

Ray

Docs ↗

Ephemeral Ray clusters on Kubernetes for distributed training and tuning.

pip install flyteplugins-ray
pip install flyteplugins-ray
Observability & Experiment Tracking

Know what your models and agents are doing

Traces that survive crashes and resumes, experiment tracking inside tasks, and agent-level observability in Grafana.

Grafana agent observability

Docs ↗

Generations, tool calls, tokens, and cost for every agent run, in Grafana.

pip install flyteplugins-agento11y
pip install flyteplugins-agento11y

MLFlow

Docs ↗

Log MLflow experiments and metrics from inside your tasks.

pip install flyteplugins-mlflow
pip install flyteplugins-mlflow

NVIDIA Nsight

Docs ↗

Profile any GPU task under Nsight Systems with a single decorator.

pip install flyteplugins-nsight
pip install flyteplugins-nsight

OpenTelemetry

Docs ↗

Distributed traces that stay intact across crashes and durable resumes.

pip install flyteplugins-otel
pip install flyteplugins-otel

Weights & Biases

Docs ↗

Experiment tracking for training runs, including distributed PyTorch.

pip install flyteplugins-wandb
pip install flyteplugins-wandb
Config & Experiment Management

Reproducible by default

Typed, versioned configuration that travels with every run.

Hydra

Docs ↗

Compose Hydra configs and launch sweeps as workflow runs.

pip install flyteplugins-hydra
pip install flyteplugins-hydra

OmegaConf

Docs ↗

Use DictConfig and ListConfig as typed task inputs and outputs.

pip install flyteplugins-omegaconf
pip install flyteplugins-omegaconf
Developer Tools

Fits your workflow

From notebooks to AI-generated code, validated before it runs.

Code Generation

Docs ↗

Generate Python from natural language and validate it with sandboxed tests.

pip install flyteplugins-codegen
pip install flyteplugins-codegen

Code sandbox

Docs ↗

Execute untrusted or generated code in isolated sandboxes.

Built In

Papermill

Docs ↗

Run parameterized Jupyter notebooks as workflow tasks.

pip install flyteplugins-papermill
pip install flyteplugins-papermill

Flyte open-source integrations.

Union.ai is built on Flyte, the open-source AI runtime trusted by thousands of companies. Every integration above ships in the open-source Flyte SDK, and the community keeps building more.

Become part of the ecosystem.

If you build a tool the AI community uses, talk to us about partnering or integrating with Union.ai.