Metadata-Version: 2.3 Name: zenml Version: 0.82.0 Summary: ZenML: Write production-ready ML code. License: Apache-2.0 Keywords: machine learning,production,pipeline,mlops,devops Author: ZenML GmbH Author-email: info@zenml.io Requires-Python: >=3.9,<3.13 Classifier: Development Status :: 4 - Beta Classifier: Intended Audience :: Developers Classifier: Intended Audience :: Science/Research Classifier: Intended Audience :: System Administrators Classifier: License :: OSI Approved :: Apache Software License Classifier: Programming Language :: Python :: 3 Classifier: Programming Language :: Python :: 3.9 Classifier: Programming Language :: Python :: 3.10 Classifier: Programming Language :: Python :: 3.11 Classifier: Programming Language :: Python :: 3.12 Classifier: Programming Language :: Python :: 3 :: Only Classifier: Topic :: Software Development :: Libraries :: Python Modules Classifier: Topic :: System :: Distributed Computing Classifier: Typing :: Typed Provides-Extra: adlfs Provides-Extra: azureml Provides-Extra: connectors-aws Provides-Extra: connectors-azure Provides-Extra: connectors-gcp Provides-Extra: connectors-kubernetes Provides-Extra: dev Provides-Extra: gcsfs Provides-Extra: s3fs Provides-Extra: sagemaker Provides-Extra: secrets-aws Provides-Extra: secrets-azure Provides-Extra: secrets-gcp Provides-Extra: secrets-hashicorp Provides-Extra: server Provides-Extra: templates Provides-Extra: terraform Provides-Extra: vertex Requires-Dist: Jinja2 ; extra == "server" Requires-Dist: adlfs (>=2021.10.0) ; extra == "adlfs" Requires-Dist: alembic (>=1.8.1,<1.9.0) Requires-Dist: aws-profile-manager (>=0.5.0) ; extra == "connectors-aws" Requires-Dist: azure-ai-ml (==1.23.1) ; extra == "azureml" Requires-Dist: azure-identity (>=1.4.0) ; extra == "secrets-azure" or extra == "connectors-azure" Requires-Dist: azure-keyvault-secrets (>=4.0.0) ; extra == "secrets-azure" Requires-Dist: azure-mgmt-containerregistry (>=10.0.0) ; extra == "connectors-azure" Requires-Dist: azure-mgmt-containerservice (>=20.0.0) ; extra == "connectors-azure" Requires-Dist: azure-mgmt-resource (>=21.0.0) ; extra == "connectors-azure" Requires-Dist: azure-mgmt-storage (>=20.0.0) ; extra == "connectors-azure" Requires-Dist: azure-storage-blob (>=12.0.0) ; extra == "connectors-azure" Requires-Dist: bandit (>=1.7.5,<2.0.0) ; extra == "dev" Requires-Dist: bcrypt (==4.0.1) Requires-Dist: boto3 (>=1.16.0) ; extra == "secrets-aws" or extra == "connectors-aws" Requires-Dist: click (>=8.0.1,<8.1.8) Requires-Dist: cloudpickle (>=2.0.0,<3) Requires-Dist: copier (>=8.1.0) ; extra == "templates" Requires-Dist: coverage[toml] (>=5.5,<6.0) ; extra == "dev" Requires-Dist: darglint (>=1.8.1,<2.0.0) ; extra == "dev" Requires-Dist: distro (>=1.6.0,<2.0.0) Requires-Dist: docker (>=7.1.0,<7.2.0) Requires-Dist: fastapi (>=0.100,<=0.115.8) ; extra == "server" Requires-Dist: gcsfs (>=2022.11.0) ; extra == "gcsfs" Requires-Dist: gitpython (>=3.1.18,<4.0.0) Requires-Dist: google-cloud-aiplatform (>=1.34.0) ; extra == "vertex" Requires-Dist: google-cloud-artifact-registry (>=1.11.3) ; extra == "connectors-gcp" Requires-Dist: google-cloud-container (>=2.21.0) ; extra == "connectors-gcp" Requires-Dist: google-cloud-pipeline-components (>=2.19.0) ; extra == "vertex" Requires-Dist: google-cloud-secret-manager (>=2.12.5) ; extra == "secrets-gcp" Requires-Dist: google-cloud-storage (>=2.9.0) ; extra == "connectors-gcp" Requires-Dist: hvac (>=0.11.2) ; extra == "secrets-hashicorp" Requires-Dist: hypothesis (>=6.43.1,<7.0.0) ; extra == "dev" Requires-Dist: importlib_metadata (<=7.0.0) ; python_version < "3.10" Requires-Dist: ipinfo (>=4.4.3) ; extra == "server" Requires-Dist: itsdangerous (>=2.2.0,<2.3.0) ; extra == "server" Requires-Dist: jinja2-time (>=0.2.0,<0.3.0) ; extra == "templates" Requires-Dist: kfp (>=2.6.0) ; extra == "vertex" Requires-Dist: kubernetes (>=18.20.0) ; extra == "connectors-kubernetes" or extra == "connectors-aws" or extra == "connectors-gcp" or extra == "connectors-azure" Requires-Dist: maison (<2.0) ; extra == "dev" Requires-Dist: mike (>=1.1.2,<2.0.0) ; extra == "dev" Requires-Dist: mkdocs (>=1.6.1,<2.0.0) ; extra == "dev" Requires-Dist: mkdocs-autorefs (>=1.4.0,<2.0.0) ; extra == "dev" Requires-Dist: mkdocs-awesome-pages-plugin (>=2.10.1,<3.0.0) ; extra == "dev" Requires-Dist: mkdocs-material (==9.6.8) ; extra == "dev" Requires-Dist: mkdocstrings[python] (>=0.28.1,<0.29.0) ; extra == "dev" Requires-Dist: mypy (==1.7.1) ; extra == "dev" Requires-Dist: orjson (>=3.10.0,<3.11.0) ; extra == "server" Requires-Dist: packaging (>=24.1) Requires-Dist: passlib[bcrypt] (>=1.7.4,<1.8.0) Requires-Dist: psutil (>=5.0.0) Requires-Dist: pydantic (>=2.0,<2.11.2) Requires-Dist: pydantic-settings Requires-Dist: pyjwt[crypto] (==2.7.*) ; extra == "server" Requires-Dist: pyment (>=0.3.3,<0.4.0) ; extra == "dev" Requires-Dist: pymysql (>=1.1.1,<1.2.0) Requires-Dist: pytest (>=7.4.0,<8.0.0) ; extra == "dev" Requires-Dist: pytest-clarity (>=1.0.1,<2.0.0) ; extra == "dev" Requires-Dist: pytest-instafail (>=0.5.0) ; extra == "dev" Requires-Dist: pytest-mock (>=3.6.1,<4.0.0) ; extra == "dev" Requires-Dist: pytest-randomly (>=3.10.1,<4.0.0) ; extra == "dev" Requires-Dist: pytest-rerunfailures (>=13.0) ; extra == "dev" Requires-Dist: pytest-split (>=0.8.1,<0.9.0) ; extra == "dev" Requires-Dist: python-dateutil (>=2.8.1,<3.0.0) Requires-Dist: python-multipart (>=0.0.9,<0.1.0) ; extra == "server" Requires-Dist: pyyaml (>=6.0.1) Requires-Dist: pyyaml-include (<2.0) ; extra == "templates" Requires-Dist: requests (>=2.27.11,<3.0.0) ; extra == "connectors-azure" Requires-Dist: rich[jupyter] (>=12.0.0) Requires-Dist: ruff (>=0.1.7) ; extra == "templates" or extra == "dev" Requires-Dist: s3fs (>=2022.11.0,!=2025.3.1) ; extra == "s3fs" Requires-Dist: sagemaker (>=2.237.3) ; extra == "sagemaker" Requires-Dist: secure (>=0.3.0,<0.4.0) ; extra == "server" Requires-Dist: setuptools Requires-Dist: sqlalchemy (>=2.0.0,<3.0.0) Requires-Dist: sqlalchemy_utils Requires-Dist: sqlmodel (==0.0.18) Requires-Dist: tldextract (>=5.1.0,<5.2.0) ; extra == "server" Requires-Dist: tox (>=3.24.3,<4.0.0) ; extra == "dev" Requires-Dist: types-Markdown (>=3.3.6,<4.0.0) ; extra == "dev" Requires-Dist: types-Pillow (>=9.2.1,<10.0.0) ; extra == "dev" Requires-Dist: types-PyMySQL (>=1.0.4,<2.0.0) ; extra == "dev" Requires-Dist: types-PyYAML (>=6.0.0,<7.0.0) ; extra == "dev" Requires-Dist: types-certifi (>=2021.10.8.0,<2022.0.0.0) ; extra == "dev" Requires-Dist: types-croniter (>=1.0.2,<2.0.0) ; extra == "dev" Requires-Dist: types-futures (>=3.3.1,<4.0.0) ; extra == "dev" Requires-Dist: types-paramiko (>=3.4.0) ; extra == "dev" Requires-Dist: types-passlib (>=1.7.7,<2.0.0) ; extra == "dev" Requires-Dist: types-protobuf (>=3.18.0,<4.0.0) ; extra == "dev" Requires-Dist: types-psutil (>=5.8.13,<6.0.0) ; extra == "dev" Requires-Dist: types-python-dateutil (>=2.8.2,<3.0.0) ; extra == "dev" Requires-Dist: types-python-slugify (>=5.0.2,<6.0.0) ; extra == "dev" Requires-Dist: types-redis (>=4.1.19,<5.0.0) ; extra == "dev" Requires-Dist: types-requests (>=2.27.11,<3.0.0) ; extra == "dev" Requires-Dist: types-setuptools (>=57.4.2,<58.0.0) ; extra == "dev" Requires-Dist: types-six (>=1.16.2,<2.0.0) ; extra == "dev" Requires-Dist: types-termcolor (>=1.1.2,<2.0.0) ; extra == "dev" Requires-Dist: typing-extensions (>=3.7.4) ; extra == "dev" Requires-Dist: uvicorn[standard] (>=0.17.5) ; extra == "server" Requires-Dist: yamlfix (>=1.16.0,<2.0.0) ; extra == "dev" Project-URL: Documentation, https://docs.zenml.io Project-URL: Homepage, https://zenml.io Project-URL: Repository, https://github.com/zenml-io/zenml Description-Content-Type: text/markdown

Beyond The Demo: Production-Grade AI Systems

ZenML brings battle-tested MLOps practices to your AI applications, handling evaluation, monitoring, and deployment at scale


ZenML Logo
[![PyPi][pypi-shield]][pypi-url] [![PyPi][pypiversion-shield]][pypi-url] [![PyPi][downloads-shield]][downloads-url] [![Contributors][contributors-shield]][contributors-url] [![License][license-shield]][license-url]
[pypi-shield]: https://img.shields.io/pypi/pyversions/zenml?color=281158 [pypi-url]: https://pypi.org/project/zenml/ [pypiversion-shield]: https://img.shields.io/pypi/v/zenml?color=361776 [downloads-shield]: https://img.shields.io/pypi/dm/zenml?color=431D93 [downloads-url]: https://pypi.org/project/zenml/ [codecov-shield]: https://img.shields.io/codecov/c/gh/zenml-io/zenml?color=7A3EF4 [codecov-url]: https://codecov.io/gh/zenml-io/zenml [contributors-shield]: https://img.shields.io/github/contributors/zenml-io/zenml?color=7A3EF4 [contributors-url]: https://github.com/zenml-io/zenml/graphs/contributors [license-shield]: https://img.shields.io/github/license/zenml-io/zenml?color=9565F6 [license-url]: https://github.com/zenml-io/zenml/blob/main/LICENSE [linkedin-shield]: https://img.shields.io/badge/-LinkedIn-black.svg?style=for-the-badge&logo=linkedin&colorB=555 [linkedin-url]: https://www.linkedin.com/company/zenml/ [twitter-shield]: https://img.shields.io/twitter/follow/zenml_io?style=for-the-badge [twitter-url]: https://twitter.com/zenml_io [slack-shield]: https://img.shields.io/badge/-Slack-black.svg?style=for-the-badge&logo=linkedin&colorB=555 [slack-url]: https://zenml.io/slack-invite [build-shield]: https://img.shields.io/github/workflow/status/zenml-io/zenml/Build,%20Lint,%20Unit%20&%20Integration%20Test/develop?logo=github&style=for-the-badge [build-url]: https://github.com/zenml-io/zenml/actions/workflows/ci.yml --- Need help with documentation? Visit our [docs site](https://docs.zenml.io) for comprehensive guides and tutorials, or browse the [SDK reference](https://sdkdocs.zenml.io/) to find specific functions and classes. ## ⭐️ Show Your Support If you find ZenML helpful or interesting, please consider giving us a star on GitHub. Your support helps promote the project and lets others know that it's worth checking out. Thank you for your support! 🌟 [![Star this project](https://img.shields.io/github/stars/zenml-io/zenml?style=social)](https://github.com/zenml-io/zenml/stargazers) ## 🀸 Quickstart [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/zenml-io/zenml/blob/main/examples/quickstart/quickstart.ipynb) [Install ZenML](https://docs.zenml.io/getting-started/installation) via [PyPI](https://pypi.org/project/zenml/). Python 3.9 - 3.12 is required: ```bash pip install "zenml[server]" notebook ``` Take a tour with the guided quickstart by running: ```bash zenml go ``` ## πŸͺ„ From Prototype to Production: AI Made Simple ### Create AI pipelines with minimal code changes ZenML is an open-source framework that handles MLOps and LLMOps for engineers scaling AI beyond prototypes. Automate evaluation loops, track performance, and deploy updates across 100s of pipelinesβ€”all while your RAG apps run like clockwork. ```python from zenml import pipeline, step @step def load_rag_documents() -> dict: # Load and chunk documents for RAG pipeline documents = extract_web_content(url="https://www.zenml.io/") return {"chunks": chunk_documents(documents)} @step def generate_embeddings(data: dict) -> None: # Generate embeddings for RAG pipeline embeddings = embed_documents(data['chunks']) return {"embeddings": embeddings} @step def index_generator( embeddings: dict, ) -> str: # Generate index for RAG pipeline index = create_index(embeddings) return index.id @pipeline def rag_pipeline() -> str: documents = load_rag_documents() embeddings = generate_embeddings(documents) index = index_generator(embeddings) return index ``` ![Running a ZenML pipeline](docs/book/.gitbook/assets/readme_simple_pipeline.gif) ### Easily provision an MLOps stack or reuse your existing infrastructure The framework is a gentle entry point for practitioners to build complex ML pipelines with little knowledge required of the underlying infrastructure complexity. ZenML pipelines can be run on AWS, GCP, Azure, Airflow, Kubeflow and even on Kubernetes without having to change any code or know underlying internals. ZenML provides different features to aid people to get started quickly on a remote setting as well. If you want to deploy a remote stack from scratch on your selected cloud provider, you can use the 1-click deployment feature either through the dashboard: ![Running a ZenML pipeline](docs/book/.gitbook/assets/one-click-deployment.gif) Or, through our CLI command: ```bash zenml stack deploy --provider aws ``` Alternatively, if the necessary pieces of infrastructure are already deployed, you can register a cloud stack seamlessly through the stack wizard: ```bash zenml stack register --provider aws ``` Read more about [ZenML stacks](https://docs.zenml.io/user-guide/production-guide/understand-stacks). ### Run workloads easily on your production infrastructure Once you have your MLOps stack configured, you can easily run workloads on it: ```bash zenml stack set python run.py ``` ```python from zenml.config import ResourceSettings, DockerSettings @step( settings={ "resources": ResourceSettings(memory="16GB", gpu_count="1", cpu_count="8"), "docker": DockerSettings(parent_image="pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime") } ) def training(...): ... ``` ![Workloads with ZenML](docs/book/.gitbook/assets/readme_compute.gif) ### Track models, pipeline, and artifacts Create a complete lineage of who, where, and what data and models are produced. You'll be able to find out who produced which model, at what time, with which data, and on which version of the code. This guarantees full reproducibility and auditability. ```python from zenml import Model @step(model=Model(name="rag_llm", tags=["staging"])) def deploy_rag(index_id: str) -> str: deployment_id = deploy_to_endpoint(index_id) return deployment_id ``` ![Exploring ZenML Models](docs/book/.gitbook/assets/readme_mcp.gif) ## πŸš€ Key LLMOps Capabilities ### Continual RAG Improvement **Build production-ready retrieval systems**
RAG Pipeline
ZenML tracks document ingestion, embedding versions, and query patterns. Implement feedback loops and: - Fix your RAG logic based on production logs - Automatically re-ingest updated documents - A/B test different embedding models - Monitor retrieval quality metrics ### Reproducible Model Fine-Tuning **Confidence in model updates**
Finetuning Pipeline
Maintain full lineage of SLM/LLM training runs: - Version training data and hyperparameters - Track performance across iterations - Automatically promote validated models - Roll back to previous versions if needed ### Purpose built for machine learning with integrations to your favorite tools While ZenML brings a lot of value out of the box, it also integrates into your existing tooling and infrastructure without you having to be locked in. ```python from bentoml._internal.bento import bento @step(on_failure=alert_slack, experiment_tracker="mlflow") def train_and_deploy(training_df: pd.DataFrame) -> bento.Bento mlflow.autolog() ... return bento ``` ![Exploring ZenML Integrations](docs/book/.gitbook/assets/readme_integrations.gif) ## πŸ”„ Your LLM Framework Isn't Enough for Production While tools like LangChain and LlamaIndex help you **build** LLM workflows, ZenML helps you **productionize** them by adding: βœ… **Artifact Tracking** - Every vector store index, fine-tuned model, and evaluation result versioned automatically βœ… **Pipeline History** - See exactly what code/data produced each version of your RAG system βœ… **Stage Promotion** - Move validated pipelines from staging β†’ production with one click ## πŸ–ΌοΈ Learning The best way to learn about ZenML is the [docs](https://docs.zenml.io/). We recommend beginning with the [Starter Guide](https://docs.zenml.io/user-guide/starter-guide) to get up and running quickly. If you are a visual learner, this 11-minute video tutorial is also a great start: [![Introductory Youtube Video](docs/book/.gitbook/assets/readme_youtube_thumbnail.png)](https://www.youtube.com/watch?v=wEVwIkDvUPs) And finally, here are some other examples and use cases for inspiration: 1. [E2E Batch Inference](examples/e2e/): Feature engineering, training, and inference pipelines for tabular machine learning. 2. [Basic NLP with BERT](examples/e2e_nlp/): Feature engineering, training, and inference focused on NLP. 3. [LLM RAG Pipeline with Langchain and OpenAI](https://github.com/zenml-io/zenml-projects/tree/main/zenml-support-agent): Using Langchain to create a simple RAG pipeline. 4. [Huggingface Model to Sagemaker Endpoint](https://github.com/zenml-io/zenml-projects/tree/main/huggingface-sagemaker): Automated MLOps on Amazon Sagemaker and HuggingFace 5. [LLMops](https://github.com/zenml-io/zenml-projects/tree/main/llm-complete-guide): Complete guide to do LLM with ZenML ## πŸ“š Learn from Books
LLM Engineer's Handbook Cover      Machine Learning Engineering with Python Cover

ZenML is featured in these comprehensive guides to modern MLOps and LLM engineering. Learn how to build production-ready machine learning systems with real-world examples and best practices. ## πŸ”‹ Deploy ZenML For full functionality ZenML should be deployed on the cloud to enable collaborative features as the central MLOps interface for teams. Read more about various deployment options [here](https://docs.zenml.io/getting-started/deploying-zenml). Or, sign up for [ZenML Pro to get a fully managed server on a free trial](https://cloud.zenml.io/?utm_source=readme&utm_medium=referral_link&utm_campaign=cloud_promotion&utm_content=signup_link). ## Use ZenML with VS Code ZenML has a [VS Code extension](https://marketplace.visualstudio.com/items?itemName=ZenML.zenml-vscode) that allows you to inspect your stacks and pipeline runs directly from your editor. The extension also allows you to switch your stacks without needing to type any CLI commands.
πŸ–₯️ VS Code Extension in Action!
ZenML Extension
## πŸ—Ί Roadmap ZenML is being built in public. The [roadmap](https://zenml.io/roadmap) is a regularly updated source of truth for the ZenML community to understand where the product is going in the short, medium, and long term. ZenML is managed by a [core team](https://zenml.io/company) of developers that are responsible for making key decisions and incorporating feedback from the community. The team oversees feedback via various channels, and you can directly influence the roadmap as follows: - Vote on your most wanted feature on our [Discussion board](https://zenml.io/discussion). - Start a thread in our [Slack channel](https://zenml.io/slack). - [Create an issue](https://github.com/zenml-io/zenml/issues/new/choose) on our GitHub repo. ## πŸ™Œ Contributing and Community We would love to develop ZenML together with our community! The best way to get started is to select any issue from the `[good-first-issue` label](https://github.com/issues?q=is%3Aopen+is%3Aissue+archived%3Afalse+user%3Azenml-io+label%3A%22good+first+issue%22) and open up a Pull Request! If you would like to contribute, please review our [Contributing Guide](CONTRIBUTING.md) for all relevant details. ## πŸ†˜ Getting Help The first point of call should be [our Slack group](https://zenml.io/slack-invite/). Ask your questions about bugs or specific use cases, and someone from the [core team](https://zenml.io/company) will respond. Or, if you prefer, [open an issue](https://github.com/zenml-io/zenml/issues/new/choose) on our GitHub repo. ## πŸ“š LLM-focused Learning Resources 1. [LL Complete Guide - Full RAG Pipeline](https://github.com/zenml-io/zenml-projects/tree/main/llm-complete-guide) - Document ingestion, embedding management, and query serving 2. [LLM Fine-Tuning Pipeline](https://github.com/zenml-io/zenml-projects/tree/main/zencoder) - From data prep to deployed model 3. [LLM Agents Example](https://github.com/zenml-io/zenml-projects/tree/main/zenml-support-agent) - Track conversation quality and tool usage ## πŸ€– AI-Friendly Documentation with llms.txt ZenML implements the llms.txt standard to make our documentation more accessible to AI assistants and LLMs. Our implementation includes: - Base documentation at [zenml.io/llms.txt](https://zenml.io/llms.txt) with core user guides - Specialized files for different documentation aspects: - [Component guides](https://zenml.io/component-guide.txt) for integration details - [How-to guides](https://zenml.io/how-to-guides.txt) for practical implementations - [Complete documentation corpus](https://zenml.io/llms-full.txt) for comprehensive access This structured approach helps AI tools better understand and utilize ZenML's documentation, enabling more accurate code suggestions and improved documentation search. ## πŸ“œ License ZenML is distributed under the terms of the Apache License Version 2.0. A complete version of the license is available in the [LICENSE](LICENSE) file in this repository. Any contribution made to this project will be licensed under the Apache License Version 2.0.

Join our Slack Slack Community and be part of the ZenML family.

Features Β· Roadmap Β· Report Bug Β· Sign up for ZenML Pro Β· Read Blog Β· Contribute to Open Source Β· Projects Showcase

πŸŽ‰ Version 0.82.0 is out. Check out the release notes here.
πŸ–₯️ Download our VS Code Extension here.