archdb

LLM architectures and pretraining recipes, with sources.

Muse-Spark-Open-816B-A42B

Meta · reasoning checkpoint

Only the post-trained multimodal checkpoint is public. Its config.json is gated, so values noted as MuseSparkTextConfig defaults come from the transformers config class declared for this checkpoint.

Open in the comparison table Hugging Face ↗

KV calculations use one sequence at 131,072 tokens and 16-bit cache elements. They include bounded sliding-window caches. This is retained cache payload, not peak prefill memory. The comparison table also offers sourced mixed-precision layouts, including quantization scales.

Release

Released

2026-10-15

Source: model-card · README.md; Muse Spark Open Model Overview; Date of Release
Date of Release | October 15th, 2026

Date the weights (or, if earlier, the technical report) were first made public, YYYY-MM-DD, as dated by the publisher. A Hugging Face repository creation date is used only when no dated publisher page exists, with a note.

License

other (Muse Model License)

Source: model-card · README.md; YAML front matter
license_name: muse-model-license

License of the released weights as declared by the publisher (model card metadata or license file), e.g. apache-2.0, mit, or the name of a custom license.

Input modalities

text + image + video + audio

Source: model-card · README.md; model summary table, Modalities
Input: Text \+ Image \+ Video \+ Audio / Output: Text

Input modalities of the released checkpoint as the publisher states them, from: text, image, video, audio, joined with ' + ' in that order ("text", "text + image + video").

Parameters

Total params

816B parameters

Source: model-card · README.md; Muse Spark Open Model Card
Muse Spark Open is an 816B total/42B active parameter open weights model with a context window of 1M tokens.

Total parameter count as reported by the publisher. See params.note for whether embeddings, MTP heads or vision towers are included.

Active params

42B parameters

Source: model-card · README.md; Muse Spark Open Model Card
Muse Spark Open is an 816B total/42B active parameter open weights model with a context window of 1M tokens.

Parameters used per token in the forward pass, as reported by the publisher.

Active non-embedding (stated)

Not disclosed

Active parameters excluding the input embedding and the output head (unembedding). Stated by the publisher, or computed from params.active by subtracting vocab × d_model for each of the embedding and output head that params.active is stated to include (basis inferred, note with the arithmetic). Not disclosed when the scope of params.active is not stated.

Counting scope

Whether 816B includes the 1.8B-param Perception Encoder is not stated.

Source: model-card · README.md; Muse Spark Open Model Overview; Type
Mixture-of-Experts Transformer, with Perception Encoder

What the publisher's counts include or exclude (MTP modules, embeddings, vision encoder, n-gram tables).

Embedding params

3.31B parameters

Inferred from the source; see the explanation below.

Source: model-card · README.md; Model Overview; Language Model
untied input/output embeddings, vocab × dim \= 202,048 × 8192 ≈ \~1.66B

2 × 202,048 × 8192 for the untied input embedding and output layer.

Parameters in input embedding and output head, as reported.

Shape

Layers

62

Source: model-card · README.md; Model Overview; Language Model
Number of Layers: 62

Number of transformer blocks in the main backbone (attention or token mixer plus FFN), excluding MTP modules.

d_model

8,192

Source: model-card · README.md; Model Overview; Language Model
Hidden Dimension: 8192

Residual stream width.

Vocab

202,048

Source: model-card · README.md; Model Overview; Vocabulary Size
Vocabulary Size | 202,048

Embedding table rows (vocab_size in config, may include padding).

Tied embeddings

No

Source: model-card · README.md; Model Overview; Language Model
untied input/output embeddings

Input embedding and output head share weights.

d_model / layers

132.129

Calculated: d_model / layers

Width-to-depth ratio, d_model divided by the number of blocks.

Attention

Token mixers

GQA, SWA

Source: code · configuration_muse_spark.py; MuseSparkTextConfig.__post_init__
Full attention on NoPE layers (every 4th, counted backward from the last), sliding otherwise — the reference config's sliding_window_pattern [w, w, w, 0].

Card: grouped-query attention.

Token-mixing layer types in the backbone, e.g. GQA, MLA, SWA, Gated DeltaNet, Mamba-2, DSA (DeepSeek sparse attention).

Layer layout

1 SWA, then 1 global, then (3 SWA : 1 global) × 15

Inferred from the source; see the explanation below.

Source: code · configuration_muse_spark.py; MuseSparkTextConfig.__post_init__; lines 153–155
"full_attention" if (self.num_hidden_layers - 1 - i) % 4 == 0 else "sliding_attention"

At 62 layers, these defaults put global attention at zero-based indices 1, 5, ..., 61 (one-based layers 2, 6, ..., 62). The checkpoint config is gated; this sequence is derived from MuseSparkTextConfig defaults, not an extracted checkpoint array.

Token-mixer order, summarized as a pattern or layer positions. The fact's optional sequence lists each mixer in order; a ratio alone does not establish the order. Auxiliary prediction modules and standalone FFNs are excluded.

Global-attention layers

16

Inferred from the source; see the explanation below.

Source: code · configuration_muse_spark.py; MuseSparkTextConfig.__post_init__
Full attention on NoPE layers (every 4th, counted backward from the last), sliding otherwise — the reference config's sliding_window_pattern [w, w, w, 0].

Layers i in 0..61 (62 layers, card) with (61 - i) % 4 == 0.

Layers whose KV cache grows with full context (dense or sparse global attention, MLA). Excludes sliding-window and linear/recurrent layers.

SWA : global

2.875

Calculated: attn.local_layers / attn.global_layers

attn.local_layers divided by attn.global_layers, the sliding-window to global interleave ratio (5 for '5 SWA : 1 global').

Sliding-window layers

46

Inferred from the source; see the explanation below.

Source: code · configuration_muse_spark.py; MuseSparkTextConfig.__post_init__
Full attention on NoPE layers (every 4th, counted backward from the last), sliding otherwise — the reference config's sliding_window_pattern [w, w, w, 0].

62 layers (card) minus 16 global layers.

Layers with a fixed attention window.

Query heads

128

Source: model-card · README.md; Model Overview; Language Model
Number of Attention Heads: 128 for Q and 16 for KV (grouped-query attention)

Query heads in the global-attention layers.

KV heads

16

Source: model-card · README.md; Model Overview; Language Model
Number of Attention Heads: 128 for Q and 16 for KV (grouped-query attention)

MuseSparkTextAttention uses the same KV head count on SWA and global layers.

Key/value heads in the global-attention layers. Not applicable to MLA, whose cache is a shared latent.

Heads × head dim

8,192

Calculated: heads × head_dim

attn.heads × attn.head_dim, the query projection width of global layers.

Q heads / KV head

8

Calculated: heads / kv_heads

attn.heads divided by attn.kv_heads in global layers.

QK head dim

64

Source: model-card · README.md; Model Overview; Language Model
Head Dimension: 64

Query/key dimension per head in global-attention layers (for MLA, nope + rope dims).

Window

2.048K tokens

Source: code · configuration_muse_spark.py; MuseSparkTextConfig
sliding_window: int | None = 2048

MuseSparkTextConfig default; the class is declared for this checkpoint (auto_docstring checkpoint=meta-models/Muse-Spark-Open-816B-A42B).

Sliding-window size of the local layers.

SWA KV heads

16

Source: model-card · README.md; Model Overview; Language Model
Number of Attention Heads: 128 for Q and 16 for KV (grouped-query attention)

Same attention class on SWA and global layers.

KV heads in sliding-window layers (stated even when equal to attn.kv_heads).

SWA query heads

128

Source: model-card · README.md; Model Overview; Language Model
Number of Attention Heads: 128 for Q and 16 for KV (grouped-query attention)

One attention class for both layer types; only the window and RoPE differ (MuseSparkTextAttention, torchtitan _muse_spark_config).

Query heads in sliding-window layers (stated even when equal to attn.heads).

SWA QK head dim

64

Source: model-card · README.md; Model Overview; Language Model
Head Dimension: 64

Same attention class on SWA and global layers.

Query/key head dimension in sliding-window layers. Value head dimensions may differ; see attn.local_v_head_dim.

Positional encoding

RoPE (SWA, θ 500000), NoPE (global)

Source: code · configuration_muse_spark.py; MuseSparkTextConfig docstring, layer_rope_theta
Defaults to the global theta everywhere except every 4th layer counted backward from the last, which is NoPE.

θ 500000 is the default rope_parameters in MuseSparkTextConfig.__post_init__, a class declared for this checkpoint.

Positional scheme in global layers: RoPE, partial RoPE, NoPE, or mixed per layer type.

SWA RoPE base

500,000

Source: code · torchtitan/models/muse_spark/flavors.py; module constants
_ROPE_THETA = 500_000.0

Also the rope_parameters default of MuseSparkTextConfig; global layers are NoPE.

RoPE base frequency of the sliding-window layers (stated even when equal to attn.rope_theta).

QK norm

Yes

Source: model-card · README.md; Model Overview; Language Model
QK-norm enabled

Weightless RMSNorm on Q and K (MuseSparkTextAttention); see attn.scale.

Queries and keys are normalized per head before the dot product.

Output gate

elementwise sigmoid gate computed from the block input, applied after a weightless per-head output RMSNorm, before o_proj

Source: code · modeling_muse_spark.py; MuseSparkTextAttention.forward
attn_output = attn_output * torch.sigmoid(self.gate_proj(hidden_states))

Card: output gate and output norm on the attention block.

Gate applied to the attention output before the output projection, with its placement.

Attention scaling

1/sqrt(head_dim) × 5.4730064863875 (Q multiplied by qk_scale_factor after QK-norm)

Source: code · configuration_muse_spark.py; MuseSparkTextConfig
qk_scale_factor: float = 5.4730064863875

MuseSparkTextConfig default; the class is declared for this checkpoint (auto_docstring checkpoint=meta-models/Muse-Spark-Open-816B-A42B). Applied in MuseSparkTextAttention.forward.

Query-key scaling when it is not 1/sqrt(head_dim), e.g. query_pre_attn_scalar, length-dependent (SSMax-style) scaling, or value scaling.

QKV bias

No

Source: code · configuration_muse_spark.py; MuseSparkTextConfig
attention_bias: bool = False

MuseSparkTextConfig default; the class is declared for this checkpoint (auto_docstring checkpoint=meta-models/Muse-Spark-Open-816B-A42B).

Query/key/value projections have bias terms.

Feed-forward

Activation

SwiGLU

Source: code · modeling_muse_spark.py; MuseSparkExperts.forward
current_hidden_states = self.act_fn(gate) * up

act_fn is hidden_act, silu by MuseSparkTextConfig default (class declared for this checkpoint).

FFN activation (SwiGLU, GeGLU, ...).

Activation clamp

none

Source: code · modeling_muse_spark.py; MuseSparkExperts.forward
current_hidden_states = self.act_fn(gate) * up

The torchtitan 816B-A42B flavor also uses plain SwiGLU; only torchtitan's Dist-MoE throughput recipes for continued training from the released checkpoint swap in ClampedSwiGLU (limit 7, α 1.702, up + 1), a different function, so it does not describe pretraining.

Clamping of GLU gate / up activations (e.g. SwiGLU limit 7), with the layers it applies to.

FFN bias

No

Inferred from the source; see the explanation below.

Source: code · modeling_muse_spark.py; MuseSparkExperts.__init__
self.gate_up_proj = nn.Parameter(torch.empty(self.num_experts, 2 * self.intermediate_dim, self.hidden_dim))

Experts hold only gate_up_proj and down_proj weights; no bias parameters.

FFN projections have bias terms.

Mixture of experts

Routed experts

256

Source: model-card · README.md; Model Overview; Language Model
Number of Experts: 256

Routed experts per MoE layer.

Experts per token

8

Source: model-card · README.md; Model Overview; Language Model
Experts Activated per Token: 8 (top-k routing)

Routed experts selected per token.

Shared experts

0

Inferred from the source; see the explanation below.

Source: code · modeling_muse_spark.py; MuseSparkMoE.__init__
self.experts = MuseSparkExperts(config) self.post_expert_proj = nn.Linear(config.moe_hidden_size, config.hidden_size, bias=False)

MuseSparkMoE has a router, pre/post projections and routed experts only; no shared-expert branch.

Always-on experts per MoE layer.

Expert dim

4,096

Source: code · configuration_muse_spark.py; MuseSparkTextConfig
moe_intermediate_size: int = 4096

MuseSparkTextConfig default; the class is declared for this checkpoint (auto_docstring checkpoint=meta-models/Muse-Spark-Open-816B-A42B). Experts run at moe_hidden_size 4096 (see moe.latent_dim).

Intermediate width of one routed expert.

Leading dense layers

0

Source: code · configuration_muse_spark.py; MuseSparkTextConfig.__post_init__
self.mlp_layer_types = ["sparse"] * self.num_hidden_layers

MuseSparkTextConfig default; the torchtitan 816B-A42B flavor also builds a MoE FFN in all 62 layers.

Number of initial blocks that use a dense FFN instead of MoE.

Router score

sigmoid

Source: code · modeling_muse_spark.py; MuseSparkTopkRouter.__init__
self.scoring_function = ACT2FN["sigmoid"]

Logits computed in FP32 and scaled by output_multiplier; an e_score_correction_bias buffer is added for top-k selection only.

Gating function over expert logits (softmax, sigmoid).

Load balancing

Not disclosed

Load-balancing method, with coefficients where disclosed.

Routed scaling factor

1

Source: code · configuration_muse_spark.py; MuseSparkTextConfig
routed_scaling_factor: float = 1.0

MuseSparkTextConfig default; the class is declared for this checkpoint (auto_docstring checkpoint=meta-models/Muse-Spark-Open-816B-A42B).

Multiplier applied to the combined routed-expert output (routed_scaling_factor).

Renormalize top-k

Yes

Source: code · modeling_muse_spark.py; MuseSparkTopkRouter.forward
top_k_weights = top_k_weights / (top_k_weights.sum(dim=-1, keepdim=True) + self.norm_topk_prob_eps)

Selected expert weights are renormalized to sum to 1.

Expert latent dim

4,096

Source: code · configuration_muse_spark.py; MuseSparkTextConfig
moe_hidden_size: int = 4096

MuseSparkTextConfig default; the class is declared for this checkpoint (auto_docstring checkpoint=meta-models/Muse-Spark-Open-816B-A42B). pre_expert_proj 8192 → 4096 (then RMSNorm) before the experts, post_expert_proj back to 8192.

Width of the projected space routed experts operate in (LatentMoE), when smaller than d_model.

Latent compression α (d_model / latent)

2

Calculated: d_model / moe.latent_dim

shape.d_model divided by moe.latent_dim, how much LatentMoE compresses the hidden state before routed experts.

Expert dim / d_model

0.5

Calculated: moe.expert_dim / d_model

moe.expert_dim divided by shape.d_model.

Sparsity (experts / top-k)

32

Calculated: experts / top_k

Routed experts divided by experts selected per token.

Active / total

0.0514706

Calculated: params.active / params.total

params.active divided by params.total.

Multi-token prediction

MTP depth

Not disclosed

Additional future tokens predicted during pretraining (0 when the model was trained without MTP).

Normalization and stability

Norm

RMSNorm (1+w gain)

Source: code · modeling_muse_spark.py; MuseSparkTextDecoderLayer.__init__
self.input_layernorm = MuseSparkTextCenteredRMSNorm(config.hidden_size, eps=config.rms_norm_eps)

Pre-attention and pre-FFN norms; post-attention norm is weightless, post-FFN norm scales the input by (1+w) before normalizing, final norm is a plain RMSNorm.

Normalization layer type in the blocks.

Norm placement

sandwich

Source: code · modeling_muse_spark.py; MuseSparkTextDecoderLayer.forward
hidden_states = self.post_attention_layernorm(hidden_states) hidden_states = self.post_attention_residual_gate(residual, hidden_states)

The post-norm output joins the residual through a learned gate (see norm.residual).

Pre-norm, post-norm, or sandwich (pre and post).

Final logit softcap

20

Source: code · configuration_muse_spark.py; MuseSparkTextConfig
final_logit_softcapping: float = 20.0

MuseSparkTextConfig default; the class is declared for this checkpoint (auto_docstring checkpoint=meta-models/Muse-Spark-Open-816B-A42B). Logits are first multiplied by output_multiplier (see norm.output_scale).

tanh soft-capping value on output logits.

Norm epsilon

1e-05

Source: code · configuration_muse_spark.py; MuseSparkTextConfig
rms_norm_eps: float = 1e-5

MuseSparkTextConfig default; the class is declared for this checkpoint (auto_docstring checkpoint=meta-models/Muse-Spark-Open-816B-A42B). Post-attention, post-FFN and post-expert norms use post_norm_eps 1e-8.

Epsilon of the block normalization layers.

Residual stream

learned per-channel residual gate: alpha·residual + beta·h, beta = sigmoid(g/0.3), alpha = sqrt(sigmoid(-g/0.3)·(1+beta))

Source: code · modeling_muse_spark.py; MuseSparkResidualGate.forward
alpha = torch.sqrt(torch.clamp(torch.sigmoid(-w) * (1.0 + beta), min=self.eps))

Temperature 0.3 is the residual_gate_temperature default of MuseSparkTextConfig (class declared for this checkpoint).

Residual design when it is not a plain sum, e.g. hyper-connections with n streams, gated residuals, depth-scaled residuals, FP32 residual accumulation.

Embedding scaling

weightless RMSNorm after embedding lookup

Source: code · modeling_muse_spark.py; MuseSparkTextNormedEmbedding.forward
return self.embed_norm(super().forward(input_ids))

Scaling or normalization applied to input embeddings (e.g. × sqrt(d_model), RMSNorm after embedding).

Output logit scaling

logits × 1/sqrt(d_model/256) = 0.1767766952966369, before the softcap

Source: code · configuration_muse_spark.py; MuseSparkTextConfig docstring, output_multiplier
Equal to `1/sqrt(hidden_size / 256)` for the released checkpoint.

Router logits are scaled by the same output_multiplier.

Scaling applied to output logits other than a softcap (e.g. muP-style width multiplier).

Context length

Pretraining context

Not disclosed

Sequence length of the main pretraining stage.

Trained context

Not disclosed

Longest sequence length trained on after context extension.

Advertised context

1M tokens

Source: model-card · README.md; Muse Spark Open Model Card
with a context window of 1M tokens

Longest context the publisher says the model supports, possibly beyond trained length via RoPE scaling.

Pretraining recipe

Pretraining tokens

Not disclosed

Tokens seen in pretraining including annealing and context extension, as reported.

Optimizer

Not disclosed

Optimizer used for the bulk of parameters.

Peak LR

Not disclosed

Peak learning rate of the main pretraining stage.

Final LR

Not disclosed

Learning rate at the end of the decay.

LR schedule

Not disclosed

Learning-rate schedule shape with phase boundaries.

Warmup

Not disclosed

Warmup length in steps or tokens.

Batch size

Not disclosed

Global batch size, including any batch-size ramp: sequences × sequence length (token total), or the token total alone when only that is stated.

Weight decay

Not disclosed

Decoupled weight decay coefficient.

Adam betas

Not disclosed

(beta1, beta2) of the Adam-family optimizer or its AdamW fallback.

Grad clip

Not disclosed

Global gradient-norm clipping threshold.

Training precision

Not disclosed

Matmul precision of pretraining (BF16, FP8 recipe, NVFP4 recipe) with the scaling granularity where disclosed. Never inferred from the released checkpoint dtype.

Data mix

text, images, audio and video; more than 100 languages

Source: model-card · README.md; Training data
Training data includes text, images, audio, and video.

Card Languages row: trained on data from more than 100 languages.

Pretraining data composition as disclosed (domain shares, synthetic fraction, repetition).

KV cache

KV bytes / token

68,480 bytes

Calculated: kv.bytes_at_context / 131072 tokens; uniform BF16, payload only; quantization overhead excluded. Global KV: 16 × (2048 × 16/8) B/entry; SWA KV: 46 × (2048 × 16/8) B/entry

Retained KV cache divided by context length, including bounded SWA caches and indexer keys. Exports use uniform BF16 at 131,072 tokens. The website can select uniform payload precision (no quantization overhead) or sourced component formats (including scales and documented padding). This is cache payload, not peak prefill memory or persistent prefix storage.

Global KV bytes / token

65,536 bytes

Calculated: 65536 B / 1 tokens (Global KV); uniform BF16, payload only; quantization overhead excluded. Global KV: 16 × (2048 × 16/8) B/entry; SWA KV: 46 × (2048 × 16/8) B/entry

Long-context global KV bytes per token, including indexer keys; excludes bounded SWA caches. Compression rates and cross-layer cache reuse are included. Exports use uniform BF16; the website's storage selector also applies here. This is the field to compare with global-KV figures in reports, using the matching storage recipe.

KV bytes / token / B active

1,630 bytes

Calculated: kv.bytes_per_token (131072 tokens) / (params.active / 1e9); uniform BF16, payload only; quantization overhead excluded. Global KV: 16 × (2048 × 16/8) B/entry; SWA KV: 46 × (2048 × 16/8) B/entry

kv.bytes_per_token divided by active parameters in billions (131,072 tokens in exports).

KV at context / B active

213,709,775 bytes

Calculated: kv.bytes_at_context (131072 tokens) / (params.active / 1e9); uniform BF16, payload only; quantization overhead excluded. Global KV: 16 × (2048 × 16/8) B/entry; SWA KV: 46 × (2048 × 16/8) B/entry

kv.bytes_at_context divided by active parameters in billions (131,072 tokens in exports).

KV at context

8,975,810,560 bytes

Calculated: 65536 B × 131072 (Global KV) + 188416 B × min(131072, 2048) (SWA KV); uniform BF16, payload only; quantization overhead excluded. Global KV: 16 × (2048 × 16/8) B/entry; SWA KV: 46 × (2048 × 16/8) B/entry

Retained KV cache for one sequence at the selected context, including SWA windows and indexer keys. Exports use uniform BF16 at 131,072 tokens. Excludes recurrent state, compressor tail state, draft/MTP caches, temporary prefill tensors, allocator metadata and persistent prefix snapshots. Does not estimate peak prefill or total serving memory.

Evaluations

AA Intelligence Index

39

Source: leaderboard · Artificial Analysis model page /models/muse-spark-open, Wayback snapshot 2026-10-19
Muse Spark Open (Xhigh) scores 39 on the Artificial Analysis Intelligence Index

Variant: Muse Spark Open (Xhigh). Index v4.3.2.

Artificial Analysis Intelligence Index v4.3.2 of the post-trained model, from a dated Wayback snapshot of its Artificial Analysis page. The note names the variant (reasoning setting).

All models · Link to this model