Kimi-Linear#

A hybrid architecture from Moonshot AI that interleaves Kimi Delta Attention (KDA) linear-attention layers with Multi-head Latent Attention (MLA) full-attention layers. Like other Mamba / linear-attention hybrids, the KDA layers keep a recurrent state cache (a convolution + delta-net state) instead of a paged key/value cache; LMCache reinterprets that state as an opaque page at registration time, so prefix caching and KV reuse work end to end. See Hybrid Attention Models for the general handling of Mamba / linear-attention models.

Validated models#

Engine documentation: Kimi-Linear in vLLM supported models (architecture KimiLinearForCausalLM).

Status: Validated with LMCache.

As a Mamba / linear-attention hybrid, Kimi-Linear needs the same settings as the other GDN hybrids: the align Mamba cache mode, prefix caching, and --separate-object-groups on the LMCache server. The chunk size is negotiated from vLLM’s unified block size N, which is model- and parallelism-specific — vLLM logs Setting attention block size to N tokens at startup. Because the KDA state is sharded across tensor-parallel ranks (while the MLA cache is not), N depends on --tensor-parallel-size; read it from the log for your own configuration:

Model

Unified block size N

GPUs (TP)

moonshotai/Kimi-Linear-48B-A3B-Instruct

944

2

Note

N scales inversely with the tensor-parallel size for this model, so changing --tensor-parallel-size changes the negotiated chunk size. The MLA cache is replicated across TP ranks (its per-rank size is fixed), while the KDA state is sharded (its per-rank size is divided by the TP degree). vLLM picks N so an attention page is at least as large as a KDA state page, so a smaller per-rank KDA state yields a smaller N — and vice versa. Concretely, going from TP=2 → TP=1 doubles the per-rank KDA state, so N doubles from 944 to 1888 and the server negotiates a compatible chunk size. Always re-read N from the startup log after changing the TP degree — do not scale it by hand.

Start the LMCache MP server (N = 944):

lmcache server --chunk-size 944 --separate-object-groups \
    --l1-size-gb 100 --eviction-policy LRU

Start vLLM with the LMCache MP connector (2 GPUs):

vllm serve moonshotai/Kimi-Linear-48B-A3B-Instruct \
    --tensor-parallel-size 2 \
    --trust-remote-code \
    --enable-prefix-caching \
    --mamba-cache-mode align \
    --kv-transfer-config \
    '{"kv_connector":"LMCacheMPConnector", "kv_role":"kv_both"}'

Why these settings:

  • --mamba-cache-mode align and --enable-prefix-caching are required. align is the only Mamba cache mode the KDA backend supports, and prefix caching must be on for LMCache to store and reuse the recurrent state.

  • --separate-object-groups (server) is required for hybrid Mamba / linear-attention models: it gives the KDA layers their own cache objects so their recurrent state is stored and loaded independently of the MLA layers.

  • --chunk-size (server) is a minimum. The connector negotiates the smallest multiple of N that is at least that value. See Chunk-Size Negotiation.

  • --trust-remote-code loads Kimi-Linear’s custom modeling code.

  • --tensor-parallel-size 2 shards the weights across two GPUs. Adjust it to your hardware — but note it changes N (see the note above).

No attention-backend or --no-disable-hybrid-kv-cache-manager flag is needed; LMCacheMPConnector advertises hybrid support and vLLM auto-selects the KDA and MLA backends. For the generic LMCache + vLLM wiring (ports, remote hosts), see Quickstart.

Status: Not validated with LMCache.

Status: Not validated with LMCache.

CacheBlend support#

Not supported: the hybrid groups’ cached pages are byte-opaque (see Caveats).

Compression support#

Method

Status

Notes

CacheGen

Not supported

Hybrid groups’ cached pages are byte-opaque.

Caveats#

  • Generation is not guaranteed bit-exact between a cached and a fresh run under concurrent load: KDA / GDN linear-attention backends do not support vLLM’s batch-invariant mode, so kernel results can vary with batch composition. Validate with a score-level comparison, not a token-level diff.

  • Cached pages for the KDA and MLA groups are byte-opaque views, so content-aware processing (CacheGen, CacheBlend) does not apply, and cache entries must not be shared across engines with different attention backends or kernel block sizes.

  • vLLM’s Mamba prefix caching in align mode is experimental upstream.