Zem: Unified Data Pipeline Framework (ZenML + NeMo Curator + DataJuicer) for multi-domain processing
Project description
🚀 Zem
Zem is a high-performance, unified data pipeline framework designed for the modern AI era. It seamlessly bridges ZenML's production-grade orchestration with specialized curation powerhouses like NVIDIA NeMo Curator and Alibaba Data-Juicer using the Model Context Protocol (MCP).
✨ Key Features
- 🏗️ Config-Driven Power: Define complex, production-ready pipelines in single YAML files.
- ⚡ True Parallel DAGs: Execute independent processing branches concurrently using a custom
ParallelLocalOrchestrator. - 🧠 Frontier LLM Integration: Smart data masking, classification, and summarization via Ollama or OpenAI.
- 📊 Deep Observability: Real-time profiling, per-tool performance metrics, and a beautiful integrated dashboard.
- 🔄 Adaptive Caching: Fine-grained, step-level cache control to optimize your development cycles.
- 🔌 Cloud Native: Native support for S3, GCS, and Parquet with seamless export to Hugging Face Hub and Vector DBs.
🏗️ Architecture
graph TD
YAML["📄 pipeline.yaml"] --> Client["🛠️ Zem CLI / Client"]
Client --> ZenML["🌀 ZenML Orchestrator"]
ZenML --> Parallel["⚡ Parallel Local Orchestrator"]
Parallel --> MCP_Bridge["🔗 MCP Bridge"]
subgraph "Specialized Servers (MCP)"
MCP_Bridge --> Nemo["🦁 NeMo Curator (GPU)"]
MCP_Bridge --> DJ["🧃 Data-Juicer"]
MCP_Bridge --> LLM["🤖 Frontier LLMs"]
MCP_Bridge --> Prof["📈 Profiler"]
end
subgraph "Storage & Sinks"
Nemo --> S3["☁️ Cloud / Parquet"]
DJ --> HF["🤗 Hugging Face"]
LLM --> VDB["🌐 Vector DB"]
end
🚀 Quick Start
1. Installation
git clone https://github.com/OAI-Labs/xfmr-zem.git
cd xfmr-zem
uv sync
2. Initialize a New Project
# Bootstrap a standalone project with a sample agent
uv run zem init my_project
cd my_project
3. Run Your First Pipeline
uv run zem run pipeline.yaml
4. Visualize & Inspect
# Open ZenML Dashboard
uv run zem dashboard
# Preview results with sampling
uv run zem preview <artifact_id> --sample --limit 5
📖 Guided Documentation
| Topic | Description | Link |
|---|---|---|
| Core Concepts | Understand the Zem architecture and MCP model. | AGENTS.md |
| Pipeline YAML | How to write and validate your pipeline configs. | Standard Example |
| Advanced Parallelism | Setup true local concurrency. | Parallel Guide |
| LLM & Sinks | Connecting to external AI stacks. | Phase 4 Demo |
🤝 Contributing
We welcome contributions! Whether it's a new MCP server, a performance fix, or a typo in the docs, feel free to open a Pull Request.
- Fork the Project
- Create your Feature Branch (
git checkout -b feature/AmazingFeature) - Commit your Changes (
git commit -m 'Add some AmazingFeature') - Push to the Branch (
git push origin feature/AmazingFeature) - Open a Pull Request
⚖️ License
Distributed under the Apache-2.0 License. See LICENSE for more information.
Built with ❤️ by the OAI-Labs Team
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file xfmr_zem-0.2.8.tar.gz.
File metadata
- Download URL: xfmr_zem-0.2.8.tar.gz
- Upload date:
- Size: 1.7 MB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.7
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
1d67bc7151f12080b13e66760af4ffd29069c4b5b026ba2146505b922bf7f03c
|
|
| MD5 |
cb9be60693bc527d84a1172d3902968c
|
|
| BLAKE2b-256 |
d9641fc8075fd4827f31ffbe6d5258f5caf4a9a940822b6b9cd7810e236e7086
|
Provenance
The following attestation bundles were made for xfmr_zem-0.2.8.tar.gz:
Publisher:
pypi-publish.yml on OAI-Labs/xfmr-zem
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
xfmr_zem-0.2.8.tar.gz -
Subject digest:
1d67bc7151f12080b13e66760af4ffd29069c4b5b026ba2146505b922bf7f03c - Sigstore transparency entry: 907872827
- Sigstore integration time:
-
Permalink:
OAI-Labs/xfmr-zem@f6862b9fd446e3de233d895c1f92176b4c75e982 -
Branch / Tag:
refs/tags/v0.2.8 - Owner: https://github.com/OAI-Labs
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
pypi-publish.yml@f6862b9fd446e3de233d895c1f92176b4c75e982 -
Trigger Event:
push
-
Statement type:
File details
Details for the file xfmr_zem-0.2.8-py3-none-any.whl.
File metadata
- Download URL: xfmr_zem-0.2.8-py3-none-any.whl
- Upload date:
- Size: 123.0 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.7
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
8a807abfb40e9cc7f0d8c334b7fbae88dda38675aceffcfadd3aec4b0c1ccc3b
|
|
| MD5 |
7337604da7f15f51dec770b6d13e5e00
|
|
| BLAKE2b-256 |
1545708a785294007e37dff422b4eaaf76cccbbd6f5e9b01b01a82c489f18807
|
Provenance
The following attestation bundles were made for xfmr_zem-0.2.8-py3-none-any.whl:
Publisher:
pypi-publish.yml on OAI-Labs/xfmr-zem
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
xfmr_zem-0.2.8-py3-none-any.whl -
Subject digest:
8a807abfb40e9cc7f0d8c334b7fbae88dda38675aceffcfadd3aec4b0c1ccc3b - Sigstore transparency entry: 907872832
- Sigstore integration time:
-
Permalink:
OAI-Labs/xfmr-zem@f6862b9fd446e3de233d895c1f92176b4c75e982 -
Branch / Tag:
refs/tags/v0.2.8 - Owner: https://github.com/OAI-Labs
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
pypi-publish.yml@f6862b9fd446e3de233d895c1f92176b4c75e982 -
Trigger Event:
push
-
Statement type: