StepFun · 2026-03 · Sparse MoE

Step-3.5-Flash

Step-3.5-Flash is a multi-head attention (MHA) transformer released by StepFun in 2026-03, with 45 layers, hidden size 4096 and a context window of 262,144 tokens.

Layer stack

Attention ×12 Sliding window ×33 · Layers 45

Key facts

FamilyStepFun
Released2026-03
Params196 B
Active11 B
Context262,144 tokens
AttentionMHA (64:64)
Layers45
Hidden4,096
Heads64
Vocab128,896
PositionRoPE
Norm
Activation
Dtypebf16
Architecture classStep3p5ForCausalLM

Architecture overview

t₁ t₂ t₃ t₄ input tokens Embedding · Vocab 128,896 → Hidden 4,096 Q1 Q2 Q3 Qn KV1 KV2 KV3 KVn Heads 64 KV heads 64 head dim 128 · RoPE Router E1 E2 E3 E4 E5 E6 E7 E8 +280… Shared expert ×1 top-8 of 288 experts · ≈ 11B params active per token × 45 transformer block Attention Feed-forward / MoE pre-norm Final norm · LM head → Vocab 128,896 p p p → next token MTP ×3 → +3 future tokens Position RoPE Dtype BF16 Context 256K tok
Drawn from the shipped config.json · 45 layers / width 4,096 / context 262,144. Original diagram by this atlas.

Attention

multi-head attention (MHA) — 64 q-heads / 64 kv-heads. RoPE θ=NaN.

Feed-forward / MoE

It is a sparse mixture-of-experts with 288 routed experts and top-8 routing; about 11B parameters are active per token.

The configuration ships 3 extra multi-token-prediction (MTP) layers used to accelerate decoding.

Field-level comparison

Compared against the previous model of the same lab; where none exists, against the structurally closest model. The ratio column is this model divided by the comparison model.

Full-field comparison vs predecessor Step-3
ModelStep-3.5-FlashStep-3Ratio
model_typestep3p5step3_vl
architecturesStep3p5ForCausalLMStep3VLForConditionalGeneration
hidden_size40967168×0.571
num_hidden_layers4561×0.738
num_attention_heads6464≈1
head_dim128256×0.500
intermediate_size1126418432×0.611
moe_intermediate_size12805120×0.250
rope_theta5000000.0×12 + 10000.0×36500000
max_position_embeddings262144
sliding_window512
vocab_size128896128815≈1
torch_dtypebfloat16bfloat16
num_nextn_predict_layers3
yarn_only_typesfull_attention×1
max_seq_len26214465536×4.00
use_qk_normtrue
moe_layers_enum3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40,41,42,43,444,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40,41,42,43,44,45,46,47,48,49,50,51,52,53,54,55,56,57,58,59
num_attention_groups81×8.00
use_moetrue
moe_num_experts28848×6.00
moe_top_k83×2.67
share_expert_dim12805120×0.250
moe_layer_offset0
moe_every_n_layer1
norm_expert_weighttruefalse

Most similar architectures

Similarity values range from 0 (no shared categorical features) to 1 (identical profiles).

Raw config fields

41 fields
architecturesStep3p5ForCausalLM
model_typestep3p5
yarn_only_typesfull_attention×1
hidden_size4096
intermediate_size11264
num_hidden_layers45
max_seq_len262144
vocab_size128896
torch_dtypebfloat16
use_qk_normtrue
moe_layers_enum3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40,41,42,43,44
num_attention_heads64
num_attention_groups8
head_dim128
use_moetrue
moe_num_experts288
moe_top_k8
moe_intermediate_size1280
share_expert_dim1280
moe_layer_offset0
moe_every_n_layer1
norm_expert_weighttrue
moe_router_activationsigmoid
moe_router_scaling_factor3
att_impl_typeGQA
tie_word_embeddingsfalse
rope_theta5000000.0×12 + 10000.0×36
use_head_wise_attn_gatetrue
sliding_window512
use_moe_router_biastrue
need_fp32_gatetrue
sinkfalse
layer_typesfull_attention×12 + sliding_attention×36
num_nextn_predict_layers3
partial_rotary_factors0.5×12 + 1.0×36
eos_token_id1×1 + 2×1 + 128007×1
bos_token_id0
swiglu_limits0.0×46 + 7×2
swiglu_limits_shared0.0×47 + 16×1
zero_centeredtrue
max_position_embeddings262144