Skip to main content

Momahub: 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 Momahub 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.4.tar.gz (60.9 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.4-py3-none-any.whl (75.4 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: momahub-0.2.4.tar.gz
  • Upload date:
  • Size: 60.9 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.4.tar.gz
Algorithm Hash digest
SHA256 d2348c6b8d17fa418dc3acce04a5ff857553addfcba6e15f176fbe6c94691061
MD5 f6fd3dff8fbddcec2a717719c03056ea
BLAKE2b-256 d8d2fec4a5665e1462847d417cd26ce9742005c63d038140776b3490f5c4f379

See more details on using hashes here.

File details

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

File metadata

  • Download URL: momahub-0.2.4-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.4-py3-none-any.whl
Algorithm Hash digest
SHA256 a4dce28543ea47eaa978790b8b323f4c271362721c530a4676762893c959a408
MD5 a7e27a1804d008401ee745144ebe3691
BLAKE2b-256 114b8a7ebc288137cbcab147a7abb79efff18f1e0d3c1e264fc8145999ac1efe

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