Axiom

A domain-agnostic platform for building federated, AI-powered operational systems.

Axiom is the intelligence substrate — LLM routing, RAG, memory, an agent fleet, federation, and a self-discovering CLI — that domain products build on. Each product brings its own knowledge, agents, and tools; Axiom provides everything underneath. A domain consumer (e.g. a scientific-facility consumer) sits on top as the first layer.

Install

pip install axiom-os-lm
axi config      # onboarding wizard: provisions a local llamafile + model
axi status      # platform health
axi chat        # interactive agent

Architecture

flowchart TB
    User["User · laptop / mobile / IDE / MCP client"]
    GW["LLM Gateway<br/>tier routing · circuit breakers · audit"]
    LLM["Providers<br/>self-hosted Qwen · Claude · OpenAI · Ollama"]
    RAG["RAG<br/>pgvector · 3-tier corpus"]
    MEM["Composition Memory<br/>CompositionService · MIRIX"]
    FLEET["Agent Fleet<br/>14 builtin agents"]
    CONN["Connectors<br/>Slack · Teams · Email · M365"]
    DATA["Data Platform<br/>Bronze / Silver / Gold"]
    FED["Federation<br/>node identity · trust · A2A · .axiompack"]
    PG["PostgreSQL + pgvector<br/>schema-per-extension (ADR-052)"]

    User --> GW
    GW --> LLM
    GW --> RAG
    GW --> MEM
    GW --> FLEET
    FLEET --> CONN
    FLEET --> DATA
    FLEET --> FED
    RAG --> PG
    MEM --> PG
    DATA --> PG

    style User fill:#e3f2fd,color:#000000,stroke:#1976d2
    style GW fill:#fff3e0,color:#000000,stroke:#f57c00
    style LLM fill:#f5f5f5,color:#000000,stroke:#616161
    style RAG fill:#e8f5e9,color:#000000,stroke:#388e3c
    style MEM fill:#e8f5e9,color:#000000,stroke:#388e3c
    style FLEET fill:#f3e5f5,color:#000000,stroke:#7b1fa2
    style CONN fill:#e0f2f1,color:#000000,stroke:#00897b
    style DATA fill:#e0f2f1,color:#000000,stroke:#00897b
    style FED fill:#fff9c4,color:#000000,stroke:#f9a825
    style PG fill:#f5f5f5,color:#000000,stroke:#616161
    linkStyle default stroke:#777777,stroke-width:2px

Nodes join the federation independently at any tier — a laptop, a private GPU server behind a VPN, or an HPC cluster for isolated workloads. A domain product (a consumer layer) sits on top, extending the platform through AEOS extensions.

What Axiom Provides

The Agent Fleet

Agents are LLM personas with deterministic guardrails; the CLI nouns are their purpose-named "arms and legs" (ADR-056 — axi hygiene, not axi tidy).

Quick Start

pip install axiom-os-lm

axi config                              # provision local model, write config
axi status                              # health dashboard
axi chat                                # multi-turn agent with tool calling + RAG

axi rag ingest ./docs/                  # index documents into the corpus
axi rag search "how does retry work?"   # hybrid search over the knowledge base

axi connect                             # set up an external connection (Slack/M365/Box/…)
axi federation init                     # create this node's identity and join a cohort
axi doctor                              # AI-assisted diagnostics

Bundled local model

axi config provisions a single-binary llamafile and a local weights file in ~/.axi/llamafile/ so the platform runs without a cloud key. Run axi config --model <name> to choose a different bundled model; the LLM Gateway routes to self-hosted, private-network, or cloud providers per request tier.

Extensions (AEOS)

Everything non-core is an extension conforming to the Agent Extension Open Standard. The CLI, MCP catalog, and agents discover commands, skills, tools, and agents from an axiom-extension.toml manifest:

[extension]
name = "my-extension"
version = "0.1.0"
description = "What it does"
license = "Apache-2.0"

[[extension.provides]]
kind = "cmd"        # cmd · agent · tool · service · adapter · skill · hook
noun = "myext"
entry = "my_extension.cli:main"
description = "My custom commands"

CLI verbs are thin wrappers over skill functions ((params, ctx) -> SkillResult, ADR-056), so the same capability is callable from the CLI, from a peer agent over A2A, and from an external harness over MCP. Discovery tiers: builtin (src/axiom/extensions/builtins/) → installed PyPI packages → user (~/.axi/extensions/). Scaffold one with axi ext init <name>; check conformance with axi ext lint.

Domain Products

Axiom is domain-agnostic. Domain products add the knowledge, agents, and tools for their field:

Development

git clone https://github.com/b-tree-labs/axiom-os.git
cd axiom-os
pip install -e ".[all]"

pytest                          # tests (TDD is the house rule)
ruff check src/ tests/          # lint
python -m build                 # wheel + sdist

See CLAUDE.md for architecture, conventions, and the load-bearing ADRs.

Documentation

Contributing

Contributions are welcome — code, extensions, docs, and good bug reports.

• CONTRIBUTING.md — setup, the house rules (TDD, AEOS, domain-agnostic core), the AI-assisted-contributions policy, and the DCO.
• GOVERNANCE.md — how decisions get made and what we optimize for.
• SUPPORT.md — where to ask questions (best-effort, volunteer-run).
• SECURITY.md — report vulnerabilities privately; supply-chain posture and coordinated disclosure.
• Code of Conduct — be excellent to each other.

License

Apache-2.0 — see LICENSE.

Acknowledgments

Developed at an academic institution and released as open source under Apache-2.0 with the institution's technology-transfer approval.

Copyright (c) 2026 The University of Texas at Austin and B-Tree Labs. Apache-2.0 licensed.