Skip to main content

Zem: Unified Data Pipeline Framework (ZenML + NeMo Curator + DataJuicer) for multi-domain processing

Project description

🚀 Zem

Version License ZenML MCP

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

📦 Data Versioning (DVC)

Zem tích hợp sẵn DVC để quản lý phiên bản dữ liệu lớn, sử dụng MinIO (S3-compatible) làm remote storage.

Cấu hình credentials

export DVC_MINIO_ENDPOINT=
export DVC_MINIO_BUCKET=
export DVC_MINIO_ACCESS_KEY=
export DVC_MINIO_SECRET_KEY=

Workflow

# Khởi tạo project với DVC + MinIO
uv run zem init my_project --dvc-remote minio

# Track dataset
cd my_project
uv run zem data add data/dataset.parquet -m "add training data v1"

# Push lên remote / Pull về
uv run zem data push
uv run zem data pull

# Kiểm tra trạng thái & lineage
uv run zem data status
uv run zem data lineage data/dataset.parquet

DVC metadata (hash, git commit) được tự động log vào ZenML artifact khi pipeline chạy, đảm bảo truy xuất đúng data version cho mỗi experiment.


📖 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.

  1. Fork the Project
  2. Create your Feature Branch (git checkout -b feature/AmazingFeature)
  3. Commit your Changes (git commit -m 'Add some AmazingFeature')
  4. Push to the Branch (git push origin feature/AmazingFeature)
  5. 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


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

xfmr_zem-0.3.11.tar.gz (658.5 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

xfmr_zem-0.3.11-py3-none-any.whl (291.9 kB view details)

Uploaded Python 3

File details

Details for the file xfmr_zem-0.3.11.tar.gz.

File metadata

  • Download URL: xfmr_zem-0.3.11.tar.gz
  • Upload date:
  • Size: 658.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for xfmr_zem-0.3.11.tar.gz
Algorithm Hash digest
SHA256 ff883acc909e5ba4b43b43768e19121b5e54a6fe82dc12103f3100da3a3ebb73
MD5 e6cf40fc9d77ca86fa003c7a19cb8d69
BLAKE2b-256 83b2d81f7c0118420f10f5824626d0e5c60df7b4dcd24be5d7979bdc8607c88b

See more details on using hashes here.

Provenance

The following attestation bundles were made for xfmr_zem-0.3.11.tar.gz:

Publisher: pypi-publish.yml on OAI-Labs/xfmr-zem

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file xfmr_zem-0.3.11-py3-none-any.whl.

File metadata

  • Download URL: xfmr_zem-0.3.11-py3-none-any.whl
  • Upload date:
  • Size: 291.9 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for xfmr_zem-0.3.11-py3-none-any.whl
Algorithm Hash digest
SHA256 586a3cdae6006d239f277bc93ce99c7e58eab69ce9da5240360b014eff197a50
MD5 e19393ab34d6c8aa6ac535e2187338b9
BLAKE2b-256 b042b4a3ed1ddf98d191936da9af5d1cf5abc3f8f6fb5b08d2f939f4d37bb91b

See more details on using hashes here.

Provenance

The following attestation bundles were made for xfmr_zem-0.3.11-py3-none-any.whl:

Publisher: pypi-publish.yml on OAI-Labs/xfmr-zem

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page