Skip to main content

PyStreamAI

Simple ML Deployment - Kubeflow + KServe Replacement

Zero YAML. No Kubernetes expertise required. Ship models to production in minutes.

Vision

Kubeflow solved the problem of running ML workloads on Kubernetes, but accumulated complexity. KServe requires infrastructure expertise. Modern ML engineers shouldn't need to become Kubernetes experts to deploy models.

PyStreamAI is designed to work without Kubernetes by defaultโ€”it runs locally with Docker, scales to cloud (AWS/GCP/Azure), and optionally uses Kubernetes if you already have it.

PyStreamAI replaces both Kubeflow + KServe with:

  • โœจ Dead-simple Python API โ€” No YAML, no manifests, no infrastructure expertise
  • ๐Ÿš€ Local-to-cloud โ€” Test locally, deploy with one command
  • ๐Ÿ’ฐ Cost visibility โ€” Real-time cost attribution per request
  • ๐ŸŒ Multi-cloud โ€” AWS, GCP, Azure, on-prem, Kubernetes (your choice)
  • ๐ŸŽฏ Modern AI stack โ€” LLMs, RAG, agents, not just classical ML
  • โšก Performance โ€” Rust core for speed + Python ergonomics

Quick Start

Installation

pip install pystreamai

Simple Deployment

from pystreamai import Platform

platform = Platform()

# Train a model
@platform.train(gpu="A100", time_limit="1h")
def train_model(data):
    model = train_bert(data)
    return model

job = train_model(data="s3://bucket/data")

# Deploy it
@platform.serve(replicas=3, gpu="L4")
def predict(x):
    return model.predict(x)

endpoint = predict()

# Use it
result = endpoint.predict({"text": "Hello world"})

That's it. No YAML. No Kubernetes. No infrastructure knowledge required.

Architecture

โ”Œโ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”
โ”‚   PyStreamAI Python API              โ”‚
โ”œโ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”ค
โ”‚  @train  @serve  @pipeline           โ”‚
โ”œโ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”ค
โ”‚   Rust Core Orchestrator             โ”‚
โ”œโ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”ค
โ”‚ Scheduler โ”‚ Executor โ”‚ Storage       โ”‚
โ”œโ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”ค
โ”‚         Backend Abstraction           โ”‚
โ””โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”˜
     โ†“ Deploy to โ†“
Local/Docker โ†’ AWS โ†’ GCP โ†’ Azure โ†’ Kubernetes (optional)

Backend Options (choose what works for you):

  • Local (default) โ€” Develop locally with Docker
  • Docker โ€” Run containerized models
  • AWS โ€” Lambda + EC2 + SageMaker integration
  • GCP โ€” Cloud Run + Compute Engine
  • Azure โ€” Container Instances + AML
  • Kubernetes โ€” If you already use K8s (but not required)

Features (v0.1)

  • Simple Python API
  • Local deployment
  • Training job submission
  • Model serving
  • Cost tracking
  • Multi-cloud backends
  • Observability (traces, metrics, logs)
  • Pipelines (DAGs + dynamic)
  • Model registry
  • Edge deployment

Development

Setup

# Install Rust
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh

# Install Python dependencies
pip install maturin pytest

# Build
maturin develop

Running Tests

pytest tests/

Roadmap

v0.1 (current)

  • Simple Python API
  • Local & AWS backend
  • Training + serving

v0.2

  • Pipelines (workflows)
  • Multi-cloud (GCP, Azure)
  • Cost tracking

v0.3

  • Model registry
  • Observability layer
  • Prompt management

v0.4+

  • RAG workflows
  • Agent framework
  • Edge deployment

License

Proprietary. ยฉ 2026 Georgi Mammen Mullassery.

Status

๐Ÿšง Active development. Not ready for production yet.

Download files

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

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

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

pystreamai-0.1.0-cp313-cp313-macosx_11_0_arm64.whl (256.4 kB view details)

Uploaded CPython 3.13macOS 11.0+ ARM64

File details

Details for the file pystreamai-0.1.0-cp313-cp313-macosx_11_0_arm64.whl.

File metadata

File hashes

Hashes for pystreamai-0.1.0-cp313-cp313-macosx_11_0_arm64.whl
Algorithm Hash digest
SHA256 818320667f0d5e1d48a5115f7d810164c0a6d1ce4afd84f8690762e2885a07f9
MD5 cdc3b026835fe1a9b6d194789b6201a6
BLAKE2b-256 2a0055ba7fcf55250855cad77fdfb6b6640b3a9e0db257079a28dbab23b9c4f7

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 Sentry Error logging StatusPage Status page