Skip to main content

Octomil — serve, deploy, and observe ML models on edge devices

Project description

Octomil

Run LLMs on your laptop, phone, or edge device. One command. OpenAI-compatible API.

CI PyPI License

What is this?

Octomil is a CLI + Python SDK that serves open-weight LLMs locally with an OpenAI-compatible API. It auto-detects your hardware, picks the fastest inference engine, and gives you a drop-in replacement for cloud API calls -- works on Mac (MLX), Linux/Windows (llama.cpp), and deploys to phones.

Quick start

curl -fsSL https://get.octomil.com | sh

The installer runs octomil setup in the background — it creates a venv, installs the best engine for your hardware, downloads a recommended model, and registers the MCP server with your AI tools (Claude Code, Cursor, VS Code, Codex CLI).

Then start serving:

octomil serve gemma-1b

That's it. You now have an OpenAI-compatible server on localhost:8080:

curl http://localhost:8080/v1/chat/completions \
  -d '{"model": "gemma-1b", "messages": [{"role": "user", "content": "Hello!"}]}'

Or use any OpenAI client library:

from openai import OpenAI

client = OpenAI(base_url="http://localhost:8080/v1", api_key="unused")
r = client.chat.completions.create(
    model="gemma-1b",
    messages=[{"role": "user", "content": "Explain quantum computing in 2 sentences."}],
)
print(r.choices[0].message.content)

Features

Auto engine selection -- benchmarks all available engines and picks the fastest:

octomil serve llama-3b
# => Detected: mlx-lm (38 tok/s), llama.cpp (29 tok/s), ollama (25 tok/s)
# => Using mlx-lm

60+ models -- Gemma, Llama, Phi, Qwen, DeepSeek, Mistral, Mixtral, and more:

octomil models                  # list all available models
octomil serve phi-mini          # Microsoft Phi-4 Mini (3.8B)
octomil serve deepseek-r1-7b    # DeepSeek R1 reasoning
octomil serve qwen3-4b          # Alibaba Qwen 3
octomil serve whisper-small     # Speech-to-text

Interactive chat -- one command from install to conversation:

octomil chat                        # auto-picks best model for your device
octomil chat qwen-coder-7b          # chat with a specific model
octomil chat llama-8b -s "You are a Python expert."

Launch coding agents -- power Codex, aider, or other agents with local inference:

octomil launch                  # pick an agent interactively
octomil launch codex            # launch OpenAI Codex CLI with local model
octomil launch codex --model codestral

Deploy to phones -- push models to iOS/Android devices:

octomil deploy gemma-1b --phone --rollout 10   # canary to 10% of devices
octomil status gemma-1b                        # monitor rollout
octomil rollback gemma-1b                      # instant rollback

Benchmark your hardware:

octomil benchmark gemma-1b
# Model: gemma-1b (4bit)
# Engine: mlx-lm
# Tokens/sec: 42.3
# Memory: 1.2 GB
# Time to first token: 89ms

MCP server for AI tools -- give Claude, Cursor, VS Code, and Codex access to local inference:

octomil mcp register                    # register with all detected AI tools
octomil mcp register --target claude    # register with Claude Code only
octomil mcp status                      # check registration status

Model conversion -- convert to CoreML (iOS) or TFLite (Android):

octomil convert model.pt --target ios,android

Multi-model serving -- load multiple models, route by request:

octomil serve --models smollm-360m,phi-mini,llama-3b

Supported engines

Engine Platform Install
MLX Apple Silicon Mac pip install 'octomil-sdk[mlx]'
llama.cpp Mac, Linux, Windows pip install 'octomil-sdk[llama]'
ONNX Runtime All platforms pip install 'octomil-sdk[onnx]'
MLC-LLM Mac, Linux, Android auto-detected
MNN All platforms auto-detected
ExecuTorch Mobile auto-detected
Whisper.cpp All platforms pip install 'octomil-sdk[whisper]'
Ollama Mac, Linux auto-detected if running

No engine installed? octomil serve tells you exactly what to install.

Supported models

Full model list (60+ models)
Model Sizes Engines
Gemma 3 1B, 4B, 12B, 27B MLX, llama.cpp, MNN, ONNX, MLC
Gemma 2 2B, 9B, 27B MLX, llama.cpp
Llama 3.2 1B, 3B MLX, llama.cpp, MNN, ONNX, MLC
Llama 3.1/3.3 8B, 70B MLX, llama.cpp
Phi-4 / Phi Mini 3.8B, 14B MLX, llama.cpp, MNN, ONNX
Qwen 2.5 1.5B, 3B, 7B MLX, llama.cpp, MNN, ONNX
Qwen 3 0.6B - 32B MLX, llama.cpp
DeepSeek R1 1.5B - 70B MLX, llama.cpp
DeepSeek V3 671B (MoE) MLX, llama.cpp
Mistral / Nemo / Small 7B, 12B, 24B MLX, llama.cpp
Mixtral 8x7B, 8x22B (MoE) MLX, llama.cpp
Qwen 2.5 Coder 1.5B, 7B MLX, llama.cpp
CodeLlama 7B, 13B, 34B MLX, llama.cpp
StarCoder2 3B, 7B, 15B MLX, llama.cpp
Falcon 3 1B, 7B, 10B MLX, llama.cpp
SmolLM 360M, 1.7B MLX, llama.cpp, MNN, ONNX
Whisper tiny - large-v3 Whisper.cpp
+ many more

Use aliases: octomil serve deepseek-r1 resolves to deepseek-r1-7b. Each model supports 4bit, 8bit, and fp16 quantization variants.

How it works

curl -fsSL https://get.octomil.com | sh
    │
    └── octomil setup (background)
         ├── 1. Find system Python with venv support
         ├── 2. Create ~/.octomil/engines/venv/
         ├── 3. Install best engine (mlx-lm on Apple Silicon, llama.cpp elsewhere)
         ├── 4. Download recommended model for your device
         └── 5. Register MCP server with AI tools (Claude, Cursor, VS Code, Codex)

octomil serve gemma-1b
    │
    ├── 1. Resolve model name → catalog lookup (aliases, quant variants)
    ├── 2. Detect engines     → MLX? llama.cpp? ONNX? Ollama running?
    ├── 3. Benchmark engines  → Run each, measure tok/s, pick fastest
    ├── 4. Download model     → HuggingFace Hub (cached after first pull)
    └── 5. Start server       → FastAPI on :8080, OpenAI-compatible API
                                 ├── POST /v1/chat/completions
                                 ├── POST /v1/completions
                                 └── GET  /v1/models

CLI reference

Command Description
octomil setup Install engine, download model, register MCP servers
octomil serve <model> Start an OpenAI-compatible inference server
octomil chat [model] Interactive chat (auto-starts server)
octomil launch [agent] Launch a coding agent with local inference
octomil models List available models
octomil benchmark <model> Benchmark inference speed on your hardware
octomil warmup Pre-download the recommended model for your device
octomil mcp register Register MCP server with AI tools
octomil mcp unregister Remove MCP server from AI tools
octomil mcp status Show MCP registration status
octomil mcp serve Start the HTTP agent server (REST + A2A)
octomil deploy <model> Deploy a model to edge devices
octomil rollback <model> Roll back a deployment
octomil convert <file> Convert model to CoreML / TFLite
octomil pull <model> Download a model
octomil push <file> Upload a model to registry
octomil status <model> Check deployment status
octomil scan <path> Security scan a model or app bundle
octomil completions Print shell completion setup instructions
octomil pair Pair with a phone for deployment
octomil dashboard Open the web dashboard
octomil login Authenticate with Octomil
octomil init Initialize an organization

vs. alternatives

Octomil Ollama llama.cpp (raw) Cloud APIs
One-command serve yes yes no (build from source) n/a
OpenAI-compatible API yes yes partial native
Auto engine selection yes (benchmarks all) no (single engine) n/a n/a
Deploy to phones yes no manual no
Fleet rollouts + rollback yes no no n/a
Model conversion (CoreML/TFLite) yes no no n/a
A/B testing yes no no no
Offline / on-device yes yes yes no
Cost per inference $0 (your hardware) $0 $0 $0.01-0.10
60+ models in catalog yes yes (different catalog) yes (manual download) varies
Python SDK yes yes community yes

SDKs

SDK Package Status Inference Engine
Python octomil-sdk (PyPI) Production (v2.10.1) MLX, llama.cpp, ONNX, MLC, ExecuTorch, Whisper, MNN, Ollama
Browser @octomil/browser (npm) Production (v1.0.0) ONNX Runtime Web (WebGPU + WASM)
iOS Swift Package Manager Production (v1.1.0) CoreML + MLX
Android Maven (GitHub Packages) Production (v1.2.0) TFLite + vendor NPU
Node @octomil/sdk (npm) WIP (v0.1.0) ONNX Runtime Node

Python SDK

For fleet management, model registry, and A/B testing:

from octomil import Octomil

client = Octomil(api_key="oct_...", org_id="org_123")

# Register and deploy a model
model = client.registry.ensure_model(name="sentiment", framework="pytorch")
client.rollouts.create(model_id=model["id"], version="1.0.0", rollout_percentage=10)

# Run an A/B test
client.experiments.create(
    name="v1-vs-v2",
    model_id=model["id"],
    control_version="1.0.0",
    treatment_version="1.1.0",
)

MCP Server & AI Tool Integration

Octomil registers as an MCP server across your AI coding tools so they can use local inference. octomil setup does this automatically, or you can run it manually:

octomil mcp register                    # Claude Code, Cursor, VS Code, Codex CLI
octomil mcp register --target cursor    # single tool
octomil mcp status                      # check what's registered
octomil mcp unregister                  # remove from all tools

HTTP Agent Server & x402 Payments

Octomil also exposes its tools over HTTP with an A2A agent card, OpenAPI docs, and optional micro-payments via the x402 protocol.

octomil mcp serve                       # start HTTP agent server on :8402
octomil mcp serve --port 9000           # custom port

# With x402 payment gating (agents pay per call)
OCTOMIL_X402_ADDRESS=0xYourWallet \
OCTOMIL_SETTLER_TOKEN=s402_... \
octomil mcp serve --x402

How it works:

  1. Agent calls an Octomil tool (e.g. /api/v1/run_inference)
  2. Server returns 402 Payment Required with x402 payment requirements
  3. Agent signs an EIP-3009 transferWithAuthorization and retries with x-payment header
  4. Server verifies the signature, serves the response, and accumulates the payment
  5. When payments reach the settlement threshold ($1 USDC by default), the batch is submitted to settle402 for on-chain settlement via Multicall3

Environment variables:

Variable Default Description
OCTOMIL_X402_ADDRESS Your wallet address (where you get paid)
OCTOMIL_X402_PRICE 1000 Price per call in base units (1000 = $0.001 USDC)
OCTOMIL_X402_NETWORK base Chain: base, ethereum, polygon, arbitrum, optimism
OCTOMIL_X402_THRESHOLD 1.0 Settlement threshold in USD
OCTOMIL_SETTLER_URL https://api.settle402.dev settle402 batch settlement endpoint
OCTOMIL_SETTLER_TOKEN settle402 API key

Requirements

  • Python 3.9+
  • At least one inference engine (see Supported engines)
  • macOS, Linux, or Windows

Contributing

See CONTRIBUTING.md.

License

MIT

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

octomil_sdk-2.13.0.tar.gz (527.3 kB view details)

Uploaded Source

Built Distribution

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

octomil_sdk-2.13.0-py3-none-any.whl (349.9 kB view details)

Uploaded Python 3

File details

Details for the file octomil_sdk-2.13.0.tar.gz.

File metadata

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

File hashes

Hashes for octomil_sdk-2.13.0.tar.gz
Algorithm Hash digest
SHA256 d8ace6216f39b5a1f8f24d65ed80e4683b529a095ca8fed0a99c4b959b3fdff3
MD5 5326b55245592ea489138c6c6b592a7f
BLAKE2b-256 39fbf568eac545a0ad8bcdaa03558c7ef7619906d87faad4624f29485a3d434e

See more details on using hashes here.

File details

Details for the file octomil_sdk-2.13.0-py3-none-any.whl.

File metadata

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

File hashes

Hashes for octomil_sdk-2.13.0-py3-none-any.whl
Algorithm Hash digest
SHA256 73e32ae31792bb3217b2a01c8ca9dda72b1445a939d5d27cb27ef8aac028d056
MD5 6b4b3fb300174ae27da8d5fe49819b3c
BLAKE2b-256 565574f580a5be877bd654d6883451ba3a5ff0d2fe9dc747061b7f25eca283b3

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