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 VRAMsyntax - 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
- User Guide for detailed usage instructions,
- Cookbook with 20+ examples.
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:
-
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.
-
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
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 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
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
300227d646b1a0f4612cfb20f8fc0f56c993282a7819a325220311e999acfa89
|
|
| MD5 |
f02bd4906cb9053de3d932540cf94516
|
|
| BLAKE2b-256 |
b00cfb22056a19917d623d253996b996f033799c1a6ad586a32762ee9baec208
|
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
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
836a1ceea42abbc242bd17751a8d8043dcccdd9929a61fc6002ae0940efc5dd2
|
|
| MD5 |
c1cace89016a4ccb903b63a4684013e8
|
|
| BLAKE2b-256 |
2a9bf3a05a0e574001a61277f460c77736cae2d7322e1809f9d8a8ab232ad660
|