# Lithos-metal: Qwen3.8-27B on 40-core M5 Max at 168 tokens/s

> Source: <https://github.com/lithos-ai/lithos-metal>
> Published: 2026-10-17 04:06:23+00:00

**lithos-metal is a fully open-source inference engine that generates Metal megakernels for Apple silicon.**
It brings low-latency LLM inference to your Mac, with local serving for coding agents and other applications.
The engine combines **layer-wise mixer megakernels**, **GPU-resident generation control**, and **DSpark
speculative decoding** to reduce execution overhead and make better use of each target-model pass.

[Quick start](#quick-start) · [How it works](#how-it-works) · [Performance](#performance) · [Documentation](#documentation)

Use an Apple silicon Mac with **macOS 26+**, a registered chip backend, and enough unified memory for the
target model, draft model, and context. The Qwen3.8-27B and Qwen3.6-35B-A3B configurations have been exercised
on a **40-core M5 Max with 48 GB of unified memory**. See [hardware support](#models-and-hardware) for other backends.

Install the precompiled package from the [Lithos Homebrew tap](https://github.com/lithos-ai/homebrew-tap):

```
brew install lithos-ai/tap/lithos-metal
lithos-metal serve --model nvidia/Qwen3.8-27B-NVFP4
```

The server downloads the target and its matching **LithosAI NVFP4 DSpark head**, prepares cached weight packs,
and selects the device's kernel recipes. Startup includes compilation and warmup. By default, DSpark proposes
seven tokens, which the target verifies together with one anchor token. Use `--no-draft` for target-only
generation or `--draft PATH_OR_HUB_ID` to select another DSpark checkpoint.

The default endpoint is `http://127.0.0.1:8000`, with a 32K context capacity. Use `--max-context` to adjust
capacity to the model and available memory. Run `lithos-metal serve --help` for all options.

Leave the server running and launch an installed client from another terminal:

```
lithos-metal opencode
lithos-metal claude
lithos-metal codex
lithos-metal hermes

# Other OpenAI-compatible clients:
lithos-metal env
lithos-metal run -- your-client
```

The launchers discover the served model and configure the child process without rewriting global configuration or changing client approval settings. Install the clients separately.

```
curl http://127.0.0.1:8000/v1/chat/completions \
  -H 'Content-Type: application/json' \
  -d '{"model":"nvidia/Qwen3.8-27B-NVFP4","messages":[{"role":"user","content":"Explain how a GPU megakernel works."}],"max_completion_tokens":256,"stream":true}'
```

The server provides **Chat Completions, Responses, and Anthropic Messages** adapters with text generation,
tool calls, and SSE streaming, plus Ollama's `/api/chat` for Ollama clients. One generation runs at a time.
Responses is stateless, and tools execute in the client. See the [serving design](https://github.com/lithos-ai/lithos-metal/blob/main/docs/design/serving.md) for
protocol support, caching, sampling, and API limits.

These target/draft pairs are selected automatically by the serving CLI. Run `lithos-metal models` to list them.

| Target model | Automatic DSpark head | 
|---|---|
| [Qwen3.8-27B-NVFP4](https://huggingface.co/nvidia/Qwen3.8-27B-NVFP4) | [LithosAI/Qwen3.8-27B-DSpark-NVFP4](https://huggingface.co/LithosAI/Qwen3.8-27B-DSpark-NVFP4) | 
| [Qwen3.8-27B-4bit](https://huggingface.co/mlx-community/Qwen3.8-27B-4bit) (affine 4-bit) | [LithosAI/Qwen3.8-27B-DSpark-NVFP4](https://huggingface.co/LithosAI/Qwen3.8-27B-DSpark-NVFP4) | 
| [Qwen3.6-35B-A3B-NVFP4](https://huggingface.co/nvidia/Qwen3.6-35B-A3B-NVFP4) | [LithosAI/Qwen3.6-35B-A3B-DSpark-NVFP4](https://huggingface.co/LithosAI/Qwen3.6-35B-A3B-DSpark-NVFP4) | 

The 35B hybrid MoE integration is experimental; its [adapter design](https://github.com/lithos-ai/lithos-metal/blob/main/docs/design/models.md#qwen-hybrid-moe-model) describes
the current numerical qualification. The repository also includes Qwen and Llama-compatible model packages;
see the [model adapter design](https://github.com/lithos-ai/lithos-metal/blob/main/docs/design/models.md) for supported structures and checkpoint conventions. Matching a model family does not imply support for every checkpoint.

| Chip backend | Current status | 
|---|---|
| M5 Max, 40 GPU cores | Measured target/draft recipes for the large Qwen models, including mixer megakernels | 
| M5 Pro, 20 GPU cores | Measured kernels and configurations for smaller models | 
| M3 Pro, 18 GPU cores | Probe-derived configuration | 
| M3 Max, 30 GPU cores | Unmeasured native fallback; DSpark serving fits through 20480 context on 36 GB | 
| M3 Ultra, 60 GPU cores | Measured profile with the Apple9 matrix tile | 
| M4 Pro, 20 GPU cores | Measured profile with the Apple9 matrix tile | 
| M4 Pro, 16 GPU cores | Unmeasured native fallback | 
| M4 Max, 32 or 40 GPU cores | Unmeasured native fallback | 
| M5 Max, 32 GPU cores | Independent, unmeasured native fallback | 

Backend selection uses the chip name, GPU family, and core count. Unmeasured backends have no automatic
megakernel fusion and keep the tensor accelerator disabled. See [chip backends](https://github.com/lithos-ai/lithos-metal/blob/main/monolith/backends/metal/README.md)
for configuration ownership and validation status.

Following the task-based execution model explored by [Mirage Persistent Kernel (MPK)](https://arxiv.org/abs/2512.22219),
lithos-metal decomposes operators into matrix tiles, attention partitions, and recurrent-state slices.
The compiler records their dependencies and generates specialized Metal code. Within a megakernel, persistent
workers—Metal threadgroups—execute a sequence of tasks, reusing local intermediates where possible and
synchronizing values shared between workers.

lithos-metal uses **layer-wise fusion rather than one whole-model megakernel**. In selected configurations,
the Gated DeltaNet (GDN) or full-attention mixer within a layer becomes a single megakernel; the layer's feed-forward network
(MLP) remains separate. Qwen3.8-27B is the case study for these two schedules:

| Region | Megakernel technique | 
|---|---|
| **GDN mixer** | Partition the recurrent state by value columns, retain each slice across input tokens, and coordinate projections, recurrence, gating, and normalization through their dependencies. | 
| **Full-attention mixer** | Partition the K/V context, accumulate online-softmax statistics locally, and merge partial results before output projection. | 
| **DSpark mixer** | Apply the same task compiler to block QKV projection, attention, reduction, output projection, and residual/normalization work. Fusion is selected by the draft recipe. | 

The runtime connects megakernels and conventional kernels through a **pre-encoded Metal indirect command buffer**.
GPU-resident state tracks token positions, accepted tokens, recurrent-state commit/rollback, and stopping,
with minimal CPU involvement. Bounded dispatches provide synchronization boundaries and opportunities for GPU sharing.

- **Data reuse:** retain normalization statistics, attention accumulators, and state slices within a worker
to reduce intermediate writes, rereads, and layout conversions.
- **Matrix acceleration:** on M5, Metal Performance Primitives expose GPU neural accelerators through inline
tensor operations, allowing matrix arithmetic and surrounding custom operations to share a shader.
- **Dynamic memory:** Apple GPUs from M3 onward allocate on-chip storage dynamically. The compiler tunes worker
count, tile geometry, and intermediate lifetimes together to balance local reuse and concurrency.

Fusion decisions use complete-region latency: the data movement saved must justify task scheduling and
synchronization costs. The [Apple GPU execution model](https://github.com/lithos-ai/lithos-metal/blob/main/docs/design/apple-gpu.md) and
[architecture](https://github.com/lithos-ai/lithos-metal/blob/main/docs/design/design.md) explain the execution constraints and tradeoffs behind these choices.

[DSpark](https://arxiv.org/abs/2607.05147) combines a small target-conditioned block drafter with a sequential
Markov head that models dependencies between proposals. The target verifies the block and commits the accepted
prefix. When several proposals are accepted, one target pass produces multiple output tokens, amortizing its
weight reads. lithos-metal also optimizes the draft itself with matrix kernels and selected mixer megakernels
to reduce the overhead of proposing each block. See the [DSpark design](https://github.com/lithos-ai/lithos-metal/blob/main/docs/design/speculative-decoding.md)
and [draft mixer design](https://github.com/lithos-ai/lithos-metal/blob/main/docs/design/mixers.md#dspark-draft-mixers).

The Qwen3.8-27B-NVFP4 case study uses a **40-core M5 Max with 48 GB of unified memory**.
A target-compute projection reaches approximately **168 tokens/s at 128-token context** and
**127 tokens/s at 32K context**. The [backend comparison](https://github.com/lithos-ai/lithos-metal/blob/46b2bc4cda57826c072d94afe90a58633aaeb6d9/docs/research/m5max-27b-backend-verification.md)
includes MLX, Ollama, and vLLM-Metal, with source data and reproduction details.

These figures assume six accepted tokens per eight-row target pass and exclude drafting, acceptance processing,
and rewind. Backend timing scopes differ, so they are not measured end-to-end generation rates or a direct
end-to-end speedup comparison. The [GDN](https://github.com/lithos-ai/lithos-metal/blob/46b2bc4cda57826c072d94afe90a58633aaeb6d9/docs/research/m5max-gdn-mixer-optimization.md),
[full-attention](https://github.com/lithos-ai/lithos-metal/blob/46b2bc4cda57826c072d94afe90a58633aaeb6d9/docs/research/m5max-27b-attention-optimization.md), and
[DSpark](https://github.com/lithos-ai/lithos-metal/blob/46b2bc4cda57826c072d94afe90a58633aaeb6d9/docs/research/m5max-27b-dspark-refinement.md) studies report the corresponding kernel and round measurements.

For a source build, install Xcode Command Line Tools and Python 3.12, then:

```
git clone https://github.com/lithos-ai/lithos-metal.git
cd lithos-metal
python3.12 -m venv .venv
source .venv/bin/activate
pip install '.[serve]'
lithos-metal --version
```

The package build compiles the native runtime and bundles the Metal sources and chip recipes.
Use `pip install -e '.[serve,dev]'` for an editable development install. The Python import namespace remains
`monolith` for compatibility. See [CONTRIBUTING.md](https://github.com/lithos-ai/lithos-metal/blob/main/CONTRIBUTING.md) for development and release instructions,
and [extension contracts](https://github.com/lithos-ai/lithos-metal/blob/main/docs/design/extensions.md) for adding models,
quantization formats, operations, drafters, or chip backends.

Input-normalization fusion is enabled by default and changes BF16 rounding. Use `--no-commute-norm` with the
generation CLI or `Session(..., commute_norm=False)` to disable it; the
[normalization design](https://github.com/lithos-ai/lithos-metal/blob/main/docs/design/mixers.md#input-normalization-fusion) describes the numerical contract.

The [design documentation](https://github.com/lithos-ai/lithos-metal/blob/main/docs/README.md) describes the current implementation and its contracts.

| Design | Contents | 
|---|---|
| [Architecture](https://github.com/lithos-ai/lithos-metal/blob/main/docs/design/design.md) | Compiler, GPU-resident runtime, state, and module boundaries | 
| [Apple GPU execution](https://github.com/lithos-ai/lithos-metal/blob/main/docs/design/apple-gpu.md) | Workers, memory, matrix operations, and synchronization | 
| [Mixer megakernels](https://github.com/lithos-ai/lithos-metal/blob/main/docs/design/mixers.md) | GDN, full attention, draft mixers, and normalization fusion | 
| [Speculative decoding](https://github.com/lithos-ai/lithos-metal/blob/main/docs/design/speculative-decoding.md) | DSpark, verification policy, and state commit/rollback | 
| [Serving](https://github.com/lithos-ai/lithos-metal/blob/main/docs/design/serving.md) | Protocol adapters, session setup, prefix caching, and streaming | 
| [Model adapters](https://github.com/lithos-ai/lithos-metal/blob/main/docs/design/models.md) | Checkpoint structures and numerical conventions | 
| [Extension contracts](https://github.com/lithos-ai/lithos-metal/blob/main/docs/design/extensions.md) | Models, formats, operations, drafters, and chip backends | 

lithos-metal is licensed under [Apache 2.0](https://github.com/lithos-ai/lithos-metal/blob/main/LICENSE). It is a standalone engine; it does not depend on the
MPK/Mirage runtime. Code adapted from MPK/Mirage, MLX, llama.cpp, and other projects retains its license headers
and attribution in [third_party/NOTICE](https://github.com/lithos-ai/lithos-metal/blob/main/third_party/NOTICE).
