Skip to main content

ZenML Header

One AI Platform From Pipelines to Agents

PyPi PyPi PyPi Contributors License

Projects • Roadmap • Changelog • Report Bug • Sign up for ZenML Pro • Blog • Docs

🎉 For the latest release, see the changelog.


ZenML is built for ML or AI Engineers working on traditional ML use-cases, LLM workflows, or agents, in a company setting.

At it's core, ZenML allows you to write workflows (pipelines) that run on any infrastructure backend (stacks). You can embed any Pythonic logic within these pipelines, like training a model, or running an agentic loop. ZenML then operationalizes your application by:

  1. Automatically containerizing and tracking your code.
  2. Tracking individual runs with metrics, logs, and metadata.
  3. Abstracting away infrastructure complexity.
  4. Integrating your existing tools and infrastructure e.g. MLflow, Langgraph, Langfuse, Sagemaker, GCP Vertex, etc.
  5. Allowing you to quickly iterate on experiments with an observable layer, in development and in production.

...amongst many other features.

ZenML is used by thousands of companies to run their AI workflows. Here are some featured ones:

Airbus     AXA     JetBrains     Rivian     WiseTech Global     Brevo
Leroy Merlin     Koble     Playtika     NIQ     Enel

(please email support@zenml.io if you want to be featured)

🚀 Get Started (5 minutes)

# Install ZenML with server capabilities
pip install "zenml[server]"  # pip install zenml will install a slimmer client

# Initialize your ZenML repository
zenml init

# Start local server or connect to a remote one
zenml login

You can then explore any of the examples in this repo. We recommend starting with the quickstart, which demonstrates core ZenML concepts: pipelines, steps, artifacts, snapshots, and deployments.

🏗️ Architecture Overview

ZenML uses a client-server architecture with an integrated web dashboard (zenml-io/zenml-dashboard):

  • Local Development: pip install "zenml[local]" - runs both client and server locally
  • Production: Deploy server separately, connect with pip install zenml + zenml login <server-url>

🎮 Demo

Here is a short demo:

Watch the video

🖼️ Resources

The best way to learn about ZenML is through our comprehensive documentation and tutorials:

📚 More examples

  1. Agent Architecture Comparison - Compare AI agents with LangGraph workflows, LiteLLM integration, and automatic visualizations via custom materializers
  2. Deploying ML Models - Deploy classical ML models as production endpoints with monitoring and versioning
  3. Deploying Agents - Document analysis service with pipelines, evaluation, and embedded web UI
  4. E2E Batch Inference - Complete MLOps pipeline with feature engineering
  5. LLM RAG Pipeline - Production RAG with evaluation loops
  6. Agentic Workflow (Deep Research) - Orchestrate your agents with ZenML
  7. Fine-tuning Pipeline - Fine-tune and deploy LLMs

🗣️ Chat With Your Pipelines: ZenML MCP Server

Stop clicking through dashboards to understand your ML workflows. The ZenML MCP Server lets you query your pipelines, analyze runs, and trigger deployments using natural language through Claude Desktop, Cursor, or any MCP-compatible client.

💬 "Which pipeline runs failed this week and why?"
📊 "Show me accuracy metrics for all my customer churn models"  
🚀 "Trigger the latest fraud detection pipeline with production data"

Quick Setup:

  1. Download the .dxt file from zenml-io/mcp-zenml
  2. Drag it into Claude Desktop settings
  3. Add your ZenML server URL and API key
  4. Start chatting with your ML infrastructure

The MCP (Model Context Protocol) integration transforms your ZenML metadata into conversational insights, making pipeline debugging and analysis as easy as asking a question. Perfect for teams who want to democratize access to ML operations without requiring dashboard expertise.

🤖 Kitaru: Replay-Based Evals for AI Agents

Running AI agents in production and wondering whether the cheaper model would have held, or whether the new prompt fixes Tuesday's failure? Kitaru is our open-source sister project for agents: traces you can run, not just read.

  • Every run is a recording — wrap your agent once (PydanticAI today), or import the Langfuse traces you already collect
  • Replay is re-execution — the run happens again against your real code, with recorded tool calls answered from the recording, so nothing touches live systems
  • Fork with one thing changed — a cheaper model, a new prompt, the fix in your working tree — and trust the diff
  • Scale the decision — freeze real traffic into cohorts, run experiments, gate CI on the result
  • Self-hosted, by design — one small server on your infrastructure; replays execute on workers in your environment

The split is clean: ZenML is for ML pipelines, Kitaru is for agents.

pip install kitaru

👉 kitaru.ai · GitHub · Docs

🎓 Books & Resources

ZenML is featured in these comprehensive guides to production AI systems.

🤝 Join ML Engineers Building the Future of AI

Contribute:

Stay Updated:

  • 🗺 Public Roadmap - See what's coming next
  • 📰 Blog - Best practices and case studies
  • 🎙 Slack - Talk with AI practitioners

❓ FAQs from ML Engineers Like You

Q: "Do I need to rewrite my agents or models to use ZenML?"

A: No. Wrap your existing code in a @step. Keep using scikit-learn, PyTorch, LangGraph, LlamaIndex, or raw API calls. ZenML orchestrates your tools, it doesn't replace them.

Q: "How is this different from LangSmith/Langfuse?"

A: They provide excellent observability for LLM applications. We orchestrate the full MLOps lifecycle for your entire AI stack. With ZenML, you manage both your classical ML models and your AI agents in one unified framework, from development and evaluation all the way to production deployment.

Q: "Can I use my existing MLflow/W&B setup?"

A: Yes! ZenML integrates with both MLflow and Weights & Biases. Your experiments, our pipelines.

Q: "Is this just MLflow with extra steps?"

A: No. MLflow tracks experiments. We orchestrate the entire development process – from training and evaluation to deployment and monitoring – for both models and agents.

Q: "How do I configure ZenML with Kubernetes?"

A: ZenML integrates with Kubernetes through the native Kubernetes orchestrator, Kubeflow, and other K8s-based orchestrators. See our Kubernetes orchestrator guide and Kubeflow guide, plus deployment documentation.

Q: "What about cost? I can't afford another platform."

A: ZenML's open-source version is free forever. You likely already have the required infrastructure (like a Kubernetes cluster and object storage). We just help you make better use of it for MLOps.

🛠 VS Code / Cursor Extension

Manage pipelines directly from your editor:

🖥️ VS Code Extension in Action!
ZenML Extension

Install from VS Code Marketplace.

📜 License

ZenML is distributed under the terms of the Apache License Version 2.0. See LICENSE for details.


Linux Foundation Silver Member      CNCF Silver Member

Release files for zenml 0.97.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for zenml 0.97.0
File Size Uploaded
zenml-0.97.0.tar.gz 6.1 MB Details

Built distribution (wheel)

Table of built distributions (wheels) for zenml 0.97.0
File Interpreter ABI Platform
zenml-0.97.0-py3-none-any.whl Python 3 none any Details

Total release size: 13.6 MB

Release files / zenml-0.97.0.tar.gz

Download URL zenml-0.97.0.tar.gz
Size 6.1 MB
Tags Source
SHA-256 checksum
How to use checksums
5eca8718781e0f73af618c2b27fb9c09481029c739942acf254016d7236e3f9c
BLAKE2b-256 checksum
How to use checksums
2b4989f36069c3d080cda922184dfeb8cf03cae3c30ff3520cba8a0561c000ad
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via twine/7.0.0 CPython/3.13.14

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

PyPI Publish Attestation

PyPI verified that this artifact, at this checksum, originated from the publisher listed below.

Signed by GitHub Actions, verified by PyPI on Sep 23, 2026.

Transparency log

Release files / zenml-0.97.0-py3-none-any.whl

Download URL zenml-0.97.0-py3-none-any.whl
Size 7.5 MB
Tags Python 3
SHA-256 checksum
How to use checksums
1ba0b6a44f6dc5ad0947c076311cc6a3fe96edfca27cd202d09b24d0c5eb6a1a
BLAKE2b-256 checksum
How to use checksums
186c430a588cd2c4bd9c54bbf16f9acaec1c9a409578509be47c5583399f1a54
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via twine/7.0.0 CPython/3.13.14

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

PyPI Publish Attestation

PyPI verified that this artifact, at this checksum, originated from the publisher listed below.

Signed by GitHub Actions, verified by PyPI on Sep 23, 2026.

Transparency log

Release history Release notifications | RSS feed

This release

0.97.0 This release

2 release files

0.96.2

2 release files

0.95.1

2 release files

0.95.0

2 release files

0.94.5

2 release files

0.94.4

2 release files

0.94.3

2 release files

0.94.1

2 release files

0.93.3

2 release files

0.93.2

2 release files

0.93.1

2 release files

0.93.0

2 release files

0.91.2

2 release files

0.91.1

2 release files

0.91.0

2 release files

0.85.0

2 release files

0.84.3

2 release files

0.84.1

2 release files

0.84.0

2 release files

0.83.1

2 release files

0.83.0

2 release files

0.82.1

2 release files

0.82.0

2 release files

0.81.0

2 release files

0.80.1

2 release files

0.80.0

2 release files

0.75.1

2 release files

0.75.0

2 release files

0.73.0

2 release files

0.72.0

2 release files

0.70.0

2 release files

0.68.1

2 release files

0.68.0

2 release files

0.67.0

2 release files

0.65.0

2 release files

0.63.0

2 release files

0.62.0

2 release files

0.60.0

2 release files

0.58.2

2 release files

0.58.0

2 release files

0.57.1

2 release files

0.56.4

2 release files

0.56.2

2 release files

0.56.1

2 release files

0.56.0

2 release files

0.55.4

2 release files

0.55.3

2 release files

0.55.1

2 release files

0.55.0

2 release files

0.54.1

2 release files

0.53.1

2 release files

0.53.0

2 release files

0.52.0

2 release files

0.51.0

2 release files

0.50.0

2 release files

0.47.0

2 release files

0.46.1

2 release files

0.45.6

2 release files

0.45.5

2 release files

0.45.4

2 release files

0.45.3

2 release files

0.45.2

2 release files

0.45.1

2 release files

0.45.0

2 release files

0.44.4

2 release files

0.44.3

2 release files

0.44.2

2 release files

0.44.1

2 release files

0.44.0

2 release files

0.43.1

2 release files

0.43.0

2 release files

0.42.2

2 release files

0.42.1

2 release files

0.42.0

2 release files

0.40.3

2 release files

0.40.1

2 release files

0.40.0

2 release files

0.39.1

2 release files

0.39.0

2 release files

0.38.0

2 release files

0.36.1

2 release files

0.36.0

2 release files

0.34.0

2 release files

0.32.1

2 release files

0.32.0

2 release files

0.31.1

2 release files

0.31.0

2 release files

0.22.0

2 release files

0.20.5

2 release files

0.20.4

2 release files

0.20.3

2 release files

0.13.1

2 release files

0.13.0

2 release files

0.11.0

2 release files

0.10.0

2 release files

0.9.0

2 release files

0.8.1

2 release files

0.8.0

2 release files

0.7.3

2 release files

0.7.2

2 release files

0.7.1

2 release files

0.7.0

2 release files

0.6.3

2 release files

0.6.2

2 release files

0.6.1

2 release files

0.6.0

2 release files

0.5.7

2 release files

0.5.6

2 release files

0.5.5

2 release files

0.5.4

2 release files

0.5.3

2 release files

0.5.2

2 release files

0.5.1

2 release files

0.5.0

2 release files

0.3.8

1 release file

0.3.7

1 release file

0.3.6

1 release file

0.3.5

1 release file

0.3.4

1 release file

0.3.3

1 release file

0.3.2

1 release file

0.3.1

1 release file

0.2.0

1 release file

0.1.5

1 release file

0.1.4

1 release file

0.1.3

1 release file

0.1.2

1 release file

0.1.1

1 release file

0.1.0

1 release file

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page