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.
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.
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.
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.
CrewAI
Docs ↗
Run CrewAI crews with durable tool calls and automatic recovery.
Deep Agents
Docs ↗
LangChain Deep Agents with checkpointed planning and sub-agent steps.
Google ADK
Docs ↗
Run Google Agent Development Kit agents on Union.ai's durable runtime.
Hermes
Docs ↗
Union.ai's native agent framework, durable by design.
LangChain
Docs ↗
Run LangChain chains and agents with durable, resumable execution.
LangGraph
Docs ↗
Persist LangGraph state graphs across failures without changing your graph code.
Mistral
Docs ↗
Durable agents on the Mistral Conversations API.
OpenAI Agents SDK
Docs ↗
Run OpenAI Agents SDK agents as durable workflows with checkpointed tool calls.
Pydantic AI
Docs ↗
Type-safe Pydantic AI agents running on a durable backend.
Frontier APIs to self-hosted
Serve open models on Union.ai-managed GPUs with the inference engine of your choice.
SGLang
Docs ↗
High-throughput LLM serving with SGLang, deployed as a managed app.
vLLM
Docs ↗
Serve open-weight LLMs with vLLM as an autoscaling Union.ai app.
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.
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.
Dask
Docs ↗
On-demand Dask clusters for distributed Python, managed per workflow.
Databricks
Docs ↗
Submit Spark jobs to Databricks from your workflows.
Hugging Face Datasets
Docs ↗
Move HF Datasets between tasks and reference Hub datasets directly.
JSONL
Docs ↗
Fast JSONL file and directory types with optional compression and Arrow.
Lance
Docs ↗
Lance columnar format for DataFrames, streamed lazily between tasks.
Pandera
Docs ↗
Schema-validated DataFrames (pandas, Polars, or PySpark) as task types.
Polars
Docs ↗
Pass Polars DataFrames and LazyFrames between tasks, backed by Parquet.
Snowflake
Docs ↗
Parameterized Snowflake queries, key-pair auth, and DataFrame results.
Spark
Docs ↗
Ephemeral Spark clusters on Kubernetes, sized per task and torn down after.
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.
Ray
Docs ↗
Ephemeral Ray clusters on Kubernetes for distributed training and tuning.
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.
MLFlow
Docs ↗
Log MLflow experiments and metrics from inside your tasks.
NVIDIA Nsight
Docs ↗
Profile any GPU task under Nsight Systems with a single decorator.
OpenTelemetry
Docs ↗
Distributed traces that stay intact across crashes and durable resumes.
Weights & Biases
Docs ↗
Experiment tracking for training runs, including distributed PyTorch.
Reproducible by default
Typed, versioned configuration that travels with every run.
Hydra
Docs ↗
Compose Hydra configs and launch sweeps as workflow runs.
OmegaConf
Docs ↗
Use DictConfig and ListConfig as typed task inputs and outputs.
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.
Papermill
Docs ↗
Run parameterized Jupyter notebooks as workflow tasks.
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.