Skip to main content

Momahub (i-grid): Hub-and-spoke distributed AI inference network.

Project description

Momahub - Hub-and-spoke distributed AI inference network

Users submit requests to a Hub, the Hub dispatches them to Agent nodes running Ollama.

Requirements

  • Python >= 3.11
  • Ollama on every agent node
  • GPU recommended (CPU-only works but will be slow)

Quick start

# Install Ollama on all GPU nodes
curl -fsSL https://ollama.com/install.sh | sh
bash scripts/pull_ollama.sh

# Create virtualenv
conda create -n moma python=3.11
conda activate moma

# Git clone
git clone https://github.com/digital-duck/momahub.py.git

# Install from source
pip install -e .

# (Re)Start the hub on a GPU node
moma hub up

# Get hub node IP address, to be used by an agent node to join the hub
hostname -I 
# hub-ip-address

# (Re)Start an agent on all GPU nodes including hub node
moma join http://<hub-ip-address>:8000

# Submit a task from any GPU node
moma submit "Explain distributed inference in two sentences" --model <model of your choice pulled before>

# Monitor Momahub status
moma status
moma agents
moma tasks
moma rewards

# Get Help
moma --help

Features

  • Hub-and-spoke dispatch — automatic agent selection by compute tier, VRAM, and model availability
  • Multi-hub clustering — peer hubs share capabilities and forward tasks across the network
  • Compute tiers — agents ranked PLATINUM / GOLD / SILVER / BRONZE by measured tokens-per-second
  • Reward ledger — tracks operator contributions (tasks completed, tokens generated, credits earned)
  • SPL integration — run structured prompt programs on the grid with ON GRID / WITH VRAM syntax
  • Streamlit dashboard — real-time overview, grid monitor, rewards, SPL runner, Text2SPL, and Paper Digest
  • CLI (moma) — full grid management from the terminal

Proven Performance

Momahub has been validated in real-world LAN environments:

  • 3-GPU Milestone (2026-03-08): Successfully deployed across 3 GPU nodes using 2 NVIDIA GTX 1080 Ti (11GB VRAM) + 1 NVIDIA GTX 1050 Ti (4GB VRAM). Achieved 100% completion rate on burst stress tests with automated agent-side queueing and hub-level load balancing.
  • Tiers: Measured between 50 and 100 TPS on benchmarked models.

Codebase layout

igrid/
  schema/      Pydantic models (enums, handshake, pulse, task, reward, cluster)
  hub/         Hub FastAPI app (db, state, dispatcher, cluster, monitor)
  agent/       Agent FastAPI app (hardware detection, LLM runner, telemetry)
  spl/         SPL adapter and runner
  cli/         moma CLI
  ui/          Streamlit app
docs/          User-Guide, SPL arXiv paper
cookbook/      Ready-to-run recipes
scripts/       Utility scripts
tests/         Unit and integration tests

Documentation

Research

Momahub - the python implementation for this upcoming arxiv paper (in preparation)

Momahub: A Prompt Compiler and Decentralized LLM Inference Network Wen G. Gong (2026)

The paper introduces two key ideas:

  1. The Prompt Compiler — reframing Text2SPL as a full compiler pipeline (front-end NL→SPL, mid-end CTE DAG optimisation, back-end model/VRAM mapping), with SPL as the intermediate representation between human intent and GPU execution. The compiler is self-hosting: it runs on the i-grid it compiles for.

  2. The Distributed Inference Runtime — Momahub as the runtime layer that abstracts distributed consumer GPUs into a programmable compute surface, analogous to the JVM or the Linux kernel for traditional computing.

Related work

  • SPL - Structured Prompt Language: arXiv:2602.21257

    Wen G. Gong. (2026). Structured Prompt Language: Declarative Context Management for LLMs. arXiv preprint arXiv:2602.21257.

    @article{gong2026spl,
      title={Structured Prompt Language: Declarative Context Management for LLMs},
      author={Gong, Wen G.},
      journal={arXiv preprint arXiv:2602.21257},
      year={2026}
    }
    
  • Geodesic Reranking: arXiv:2602.15860

    Wen G. Gong. (2026). Reranker Optimization via Geodesic Distances on k-NN Manifolds. arXiv preprint arXiv:2602.15860.

    @article{gong2026geodesic,
      title={Reranker Optimization via Geodesic Distances on k-NN Manifolds},
      author={Gong, Wen G.},
      journal={arXiv preprint arXiv:2602.15860},
      year={2026}
    }
    

Contributing

We welcome contributions! Please see CONTRIBUTING.md for details on our development workflow and coding standards.

License

Apache 2.0

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

momahub-0.2.3.tar.gz (61.0 kB view details)

Uploaded Source

Built Distribution

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

momahub-0.2.3-py3-none-any.whl (75.4 kB view details)

Uploaded Python 3

File details

Details for the file momahub-0.2.3.tar.gz.

File metadata

  • Download URL: momahub-0.2.3.tar.gz
  • Upload date:
  • Size: 61.0 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.14

File hashes

Hashes for momahub-0.2.3.tar.gz
Algorithm Hash digest
SHA256 300227d646b1a0f4612cfb20f8fc0f56c993282a7819a325220311e999acfa89
MD5 f02bd4906cb9053de3d932540cf94516
BLAKE2b-256 b00cfb22056a19917d623d253996b996f033799c1a6ad586a32762ee9baec208

See more details on using hashes here.

File details

Details for the file momahub-0.2.3-py3-none-any.whl.

File metadata

  • Download URL: momahub-0.2.3-py3-none-any.whl
  • Upload date:
  • Size: 75.4 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.14

File hashes

Hashes for momahub-0.2.3-py3-none-any.whl
Algorithm Hash digest
SHA256 836a1ceea42abbc242bd17751a8d8043dcccdd9929a61fc6002ae0940efc5dd2
MD5 c1cace89016a4ccb903b63a4684013e8
BLAKE2b-256 2a9bf3a05a0e574001a61277f460c77736cae2d7322e1809f9d8a8ab232ad660

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