Skip to main content

Automatic SageMaker Pipeline Generation from DAG Specifications

Project description

Cursus: Automatic SageMaker Pipeline Generation

PyPI version Python 3.8+ License: MIT

Transform pipeline graphs into production-ready SageMaker pipelines automatically.

Cursus is an intelligent pipeline generation system that automatically creates complete SageMaker pipelines from user-provided pipeline graphs. Simply define your ML workflow as a graph structure, and Cursus handles all the complex SageMaker implementation details, dependency resolution, and configuration management automatically.

🚀 Quick Start

Installation

# Core installation
pip install cursus

# With ML frameworks
pip install cursus[pytorch,xgboost]

# Full installation with all features
pip install cursus[all]

30-Second Example

from cursus.core import compile_dag_to_pipeline
from cursus.api import PipelineDAG

# Create a simple DAG
dag = PipelineDAG()
dag.add_node("CradleDataLoading")
dag.add_node("TabularPreprocessing") 
dag.add_node("XGBoostTraining")
dag.add_edge("CradleDataLoading", "TabularPreprocessing")
dag.add_edge("TabularPreprocessing", "XGBoostTraining")

# Compile to SageMaker pipeline automatically
pipeline = compile_dag_to_pipeline(dag, pipeline_name="fraud-detection")
pipeline.start()  # Deploy and run!

Command Line Interface

# Generate a new project
cursus init --template xgboost --name fraud-detection

# Validate your DAG
cursus validate my_dag.py

# Compile to SageMaker pipeline
cursus compile my_dag.py --name my-pipeline --output pipeline.json

✨ Key Features

🎯 Graph-to-Pipeline Automation

  • Input: Simple pipeline graph with step types and connections
  • Output: Complete SageMaker pipeline with all dependencies resolved
  • Magic: Intelligent analysis of graph structure with automatic step builder selection

10x Faster Development

  • Before: 2-4 weeks of manual SageMaker configuration
  • After: 10-30 minutes from graph to working pipeline
  • Result: 95% reduction in development time

🧠 Intelligent Dependency Resolution

  • Automatic step connections and data flow
  • Smart configuration matching and validation
  • Type-safe specifications with compile-time checks
  • Semantic compatibility analysis

🛡️ Production Ready

  • Built-in quality gates and validation
  • Enterprise governance and compliance
  • Comprehensive error handling and debugging
  • 98% complete with 1,650+ lines of complex code eliminated

📊 Proven Results

Based on production deployments across enterprise environments:

Component Code Reduction Lines Eliminated Key Benefit
Processing Steps 60% 400+ lines Automatic input/output resolution
Training Steps 60% 300+ lines Intelligent hyperparameter handling
Model Steps 47% 380+ lines Streamlined model creation
Registration Steps 66% 330+ lines Simplified deployment workflows
Overall System ~55% 1,650+ lines Intelligent automation

🏗️ Architecture

Cursus follows a sophisticated layered architecture:

  • 🎯 User Interface: Fluent API and Pipeline DAG for intuitive construction
  • 🧠 Intelligence Layer: Smart proxies with automatic dependency resolution
  • 🏗️ Orchestration: Pipeline assembler and compiler for DAG-to-template conversion
  • 📚 Registry Management: Multi-context coordination with lifecycle management
  • 🔗 Dependency Resolution: Intelligent matching with semantic compatibility
  • 📋 Specification Layer: Comprehensive step definitions with quality gates

📚 Usage Examples

Basic Pipeline

from cursus.core import compile_dag_to_pipeline
from cursus.api import PipelineDAG

# Create DAG
dag = PipelineDAG()
dag.add_node("CradleDataLoading")
dag.add_node("XGBoostTraining")
dag.add_edge("CradleDataLoading", "XGBoostTraining")

# Compile to SageMaker pipeline
pipeline = compile_dag_to_pipeline(dag, pipeline_name="my-ml-pipeline")

Advanced Configuration

from cursus.core import compile_dag_to_pipeline
from cursus.api import PipelineDAG

# Create DAG with more complex workflow
dag = PipelineDAG()
dag.add_node("CradleDataLoading")
dag.add_node("TabularPreprocessing")
dag.add_node("XGBoostTraining")
dag.add_node("XGBoostModelEval")
dag.add_edge("CradleDataLoading", "TabularPreprocessing")
dag.add_edge("TabularPreprocessing", "XGBoostTraining")
dag.add_edge("XGBoostTraining", "XGBoostModelEval")

# Compile with custom configuration
pipeline = compile_dag_to_pipeline(
    dag=dag,
    pipeline_name="advanced-ml-pipeline",
    config_path="config.yaml"
)

Using the Compiler Class

from cursus.core import PipelineDAGCompiler
from cursus.api import PipelineDAG

# Create DAG
dag = PipelineDAG()
dag.add_node("TabularPreprocessing")
dag.add_node("XGBoostTraining")
dag.add_edge("TabularPreprocessing", "XGBoostTraining")

# Use compiler for more control
compiler = PipelineDAGCompiler()
pipeline = compiler.compile(dag, pipeline_name="my-pipeline")

🔧 Installation Options

Core Installation

pip install cursus

Includes basic DAG compilation and SageMaker integration.

Framework-Specific

pip install cursus[pytorch]    # PyTorch Lightning models
pip install cursus[xgboost]    # XGBoost training pipelines  
pip install cursus[nlp]        # NLP models and processing
pip install cursus[processing] # Advanced data processing

Development

pip install cursus[dev]        # Development tools
pip install cursus[docs]       # Documentation tools
pip install cursus[all]        # Everything included

🎯 Who Should Use Cursus?

Data Scientists & ML Practitioners

  • Focus on model development, not infrastructure complexity
  • Rapid experimentation with 10x faster iteration
  • Business-focused interface eliminates SageMaker expertise requirements

Platform Engineers & ML Engineers

  • 60% less code to maintain and debug
  • Specification-driven architecture prevents common errors
  • Universal patterns enable faster team onboarding

Organizations

  • Accelerated innovation with faster pipeline development
  • Reduced technical debt through clean architecture
  • Built-in governance and compliance frameworks

📖 Documentation

📚 Complete Documentation Hub

Your gateway to all Cursus documentation - start here for comprehensive navigation

Knowledge Management Philosophy

  • Zettelkasten Principles - The knowledge management principles behind our slipbox documentation system, explaining how we organize and connect information for maximum discoverability and organic growth

Core Documentation

  • Developer Guide - Comprehensive guide for developing new pipeline steps and extending Cursus
  • Design Documentation - Detailed architectural documentation and design principles
  • Pipeline Catalog - Comprehensive collection of prebuilt pipeline templates organized by framework and task
  • API Reference - Detailed API documentation including core, api, steps, and other components
  • Examples - Ready-to-use pipeline blueprints and examples

Quick Links

🤝 Contributing

We welcome contributions! See our Developer Guide for comprehensive details on:

For architectural insights and design decisions, see the Design Documentation.

📄 License

This project is licensed under the MIT License - see the LICENSE file for details.

🔗 Links


Cursus: Making SageMaker pipeline development 10x faster through intelligent automation. 🚀

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

cursus-1.3.8.tar.gz (705.0 kB view details)

Uploaded Source

Built Distribution

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

cursus-1.3.8-py3-none-any.whl (927.9 kB view details)

Uploaded Python 3

File details

Details for the file cursus-1.3.8.tar.gz.

File metadata

  • Download URL: cursus-1.3.8.tar.gz
  • Upload date:
  • Size: 705.0 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.7

File hashes

Hashes for cursus-1.3.8.tar.gz
Algorithm Hash digest
SHA256 3e315722feda0d9af93e1285b6bb30564fb000b239541f65bc2ebb3ea6f538a8
MD5 479137f15e6811343c03d7074e49615d
BLAKE2b-256 829c6f1d38be6a4198116524ee1333602b78c136a73ed12f128fe2107cb096ee

See more details on using hashes here.

File details

Details for the file cursus-1.3.8-py3-none-any.whl.

File metadata

  • Download URL: cursus-1.3.8-py3-none-any.whl
  • Upload date:
  • Size: 927.9 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.7

File hashes

Hashes for cursus-1.3.8-py3-none-any.whl
Algorithm Hash digest
SHA256 db2b120156a43fd3e3da9c1c7b20a5d6a9155835532c857895ed655005e570a9
MD5 4695861d2e613273345eeb3a90d9ed38
BLAKE2b-256 d7179d56c1585a52348e5d98e08b60fffb71196c83c09e9a89b94658be62911c

See more details on using hashes here.

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