N Neurarch Architectures Models Checks Data Docs Open the app

Models / mimo_v2

MiMo-V2.5

Reconstructed from its own config.json with no weights read. 323K downloads on Hugging Face.

Our count against the checkpoint

The left number comes from the graph. The right one is the number of scalars in the published weight files. Nothing on this page was tuned to make them agree.

Derived from structure
316B
315,696,050,688 parameters
In the published checkpoint
311B
310,775,040,000 scalars · safetensors.total, read 2026-07-09
Delta
+1.58%

quantized This checkpoint is stored quantized (fp8). The tensor count in the file counts stored elements under a packing scheme, not logical parameters, so the two numbers below are not measuring the same thing in either direction.

What it costs to run

Cost is a roofline estimate on the priced GPU for 10 epochs at batch 32 over 50,000 samples (assumed; no dataset attached). GPU fit is fp32 weights plus gradients plus two Adam moments (16 bytes per parameter) with 1.3x headroom; activations are not included and grow with batch size.

Layers
300
Will it forward-pass
Yes
Priced on
A10G (24GB)
Est. one run
$24900539.68
CardMemory
T4 (16GB)weights + activationsdoes not fit
A100 (40GB)weights + activationsdoes not fit
H100 (80GB)weights + activationsdoes not fit

Structure

304 nodes. Output shapes are propagated from the input shape, batch dimension excluded.

LayerTypeOutput shape
1InputInput1 × 1048576
2EmbeddingEmbedding1 × 1048576 × 4096
3RoPERoPE1 × 1048576 × 4096
4Vision inputInput3 × 224 × 224
5PatchEmbedPatch Embed196 × 1280
6Patch_Position_EmbeddingLearned Pos Embed196 × 1280
7Vision encoder (internals not in config)Projection196 × 4096
8Vision tokensReshape1 × 196 × 4096
9Audio inputInput1 × 128 × 1
10Audio feature embedProjection1 × 128 × 4096
11Audio encoder (internals not in config)Projection1 × 128 × 4096
12Audio tokensReshape1 × 256 × 4096
13Multimodal fusion (concat tokens)Concatenate1 × 1049028 × 4096
14RMSNorm_1_1RMSNorm1 × 1049028 × 4096
15Attention_1Grouped Query Attn1 × 1049028 × 4096
16Add_1_attnAdd1 × 1049028 × 4096
17RMSNorm_1_2RMSNorm1 × 1049028 × 4096
18MoE_1MoE Layer1 × 1049028 × 4096
19Add_1_ffnAdd1 × 1049028 × 4096
20RMSNorm_2_1RMSNorm1 × 1049028 × 4096
21Attention_2Grouped Query Attn1 × 1049028 × 4096
22Add_2_attnAdd1 × 1049028 × 4096
23RMSNorm_2_2RMSNorm1 × 1049028 × 4096
24MoE_2MoE Layer1 × 1049028 × 4096
25Add_2_ffnAdd1 × 1049028 × 4096
26RMSNorm_3_1RMSNorm1 × 1049028 × 4096
27Attention_3Grouped Query Attn1 × 1049028 × 4096
28Add_3_attnAdd1 × 1049028 × 4096
29RMSNorm_3_2RMSNorm1 × 1049028 × 4096
30MoE_3MoE Layer1 × 1049028 × 4096
31Add_3_ffnAdd1 × 1049028 × 4096
32RMSNorm_4_1RMSNorm1 × 1049028 × 4096
33Attention_4Grouped Query Attn1 × 1049028 × 4096
34Add_4_attnAdd1 × 1049028 × 4096
35RMSNorm_4_2RMSNorm1 × 1049028 × 4096
36MoE_4MoE Layer1 × 1049028 × 4096
37Add_4_ffnAdd1 × 1049028 × 4096
38RMSNorm_5_1RMSNorm1 × 1049028 × 4096
39Attention_5Grouped Query Attn1 × 1049028 × 4096
40Add_5_attnAdd1 × 1049028 × 4096
41RMSNorm_5_2RMSNorm1 × 1049028 × 4096
42MoE_5MoE Layer1 × 1049028 × 4096
43Add_5_ffnAdd1 × 1049028 × 4096
44RMSNorm_6_1RMSNorm1 × 1049028 × 4096
45Attention_6Grouped Query Attn1 × 1049028 × 4096
46Add_6_attnAdd1 × 1049028 × 4096
47RMSNorm_6_2RMSNorm1 × 1049028 × 4096
48MoE_6MoE Layer1 × 1049028 × 4096
49Add_6_ffnAdd1 × 1049028 × 4096
50RMSNorm_7_1RMSNorm1 × 1049028 × 4096
51Attention_7Grouped Query Attn1 × 1049028 × 4096
52Add_7_attnAdd1 × 1049028 × 4096
53RMSNorm_7_2RMSNorm1 × 1049028 × 4096
54MoE_7MoE Layer1 × 1049028 × 4096
55Add_7_ffnAdd1 × 1049028 × 4096
56RMSNorm_8_1RMSNorm1 × 1049028 × 4096
57Attention_8Grouped Query Attn1 × 1049028 × 4096
58Add_8_attnAdd1 × 1049028 × 4096
59RMSNorm_8_2RMSNorm1 × 1049028 × 4096
60MoE_8MoE Layer1 × 1049028 × 4096
61Add_8_ffnAdd1 × 1049028 × 4096
62RMSNorm_9_1RMSNorm1 × 1049028 × 4096
63Attention_9Grouped Query Attn1 × 1049028 × 4096
64Add_9_attnAdd1 × 1049028 × 4096
65RMSNorm_9_2RMSNorm1 × 1049028 × 4096
66MoE_9MoE Layer1 × 1049028 × 4096
67Add_9_ffnAdd1 × 1049028 × 4096
68RMSNorm_10_1RMSNorm1 × 1049028 × 4096
69Attention_10Grouped Query Attn1 × 1049028 × 4096
70Add_10_attnAdd1 × 1049028 × 4096
71RMSNorm_10_2RMSNorm1 × 1049028 × 4096
72MoE_10MoE Layer1 × 1049028 × 4096
73Add_10_ffnAdd1 × 1049028 × 4096
74RMSNorm_11_1RMSNorm1 × 1049028 × 4096
75Attention_11Grouped Query Attn1 × 1049028 × 4096
76Add_11_attnAdd1 × 1049028 × 4096
77RMSNorm_11_2RMSNorm1 × 1049028 × 4096
78MoE_11MoE Layer1 × 1049028 × 4096
79Add_11_ffnAdd1 × 1049028 × 4096
80RMSNorm_12_1RMSNorm1 × 1049028 × 4096
81Attention_12Grouped Query Attn1 × 1049028 × 4096
82Add_12_attnAdd1 × 1049028 × 4096
83RMSNorm_12_2RMSNorm1 × 1049028 × 4096
84MoE_12MoE Layer1 × 1049028 × 4096
85Add_12_ffnAdd1 × 1049028 × 4096
86RMSNorm_13_1RMSNorm1 × 1049028 × 4096
87Attention_13Grouped Query Attn1 × 1049028 × 4096
88Add_13_attnAdd1 × 1049028 × 4096
89RMSNorm_13_2RMSNorm1 × 1049028 × 4096
90MoE_13MoE Layer1 × 1049028 × 4096
91Add_13_ffnAdd1 × 1049028 × 4096
92RMSNorm_14_1RMSNorm1 × 1049028 × 4096
93Attention_14Grouped Query Attn1 × 1049028 × 4096
94Add_14_attnAdd1 × 1049028 × 4096
95RMSNorm_14_2RMSNorm1 × 1049028 × 4096
96MoE_14MoE Layer1 × 1049028 × 4096
97Add_14_ffnAdd1 × 1049028 × 4096
98RMSNorm_15_1RMSNorm1 × 1049028 × 4096
99Attention_15Grouped Query Attn1 × 1049028 × 4096
100Add_15_attnAdd1 × 1049028 × 4096
101RMSNorm_15_2RMSNorm1 × 1049028 × 4096
102MoE_15MoE Layer1 × 1049028 × 4096
103Add_15_ffnAdd1 × 1049028 × 4096
104RMSNorm_16_1RMSNorm1 × 1049028 × 4096
105Attention_16Grouped Query Attn1 × 1049028 × 4096
106Add_16_attnAdd1 × 1049028 × 4096
107RMSNorm_16_2RMSNorm1 × 1049028 × 4096
108MoE_16MoE Layer1 × 1049028 × 4096
109Add_16_ffnAdd1 × 1049028 × 4096
110RMSNorm_17_1RMSNorm1 × 1049028 × 4096
111Attention_17Grouped Query Attn1 × 1049028 × 4096
112Add_17_attnAdd1 × 1049028 × 4096
113RMSNorm_17_2RMSNorm1 × 1049028 × 4096
114MoE_17MoE Layer1 × 1049028 × 4096
115Add_17_ffnAdd1 × 1049028 × 4096
116RMSNorm_18_1RMSNorm1 × 1049028 × 4096
117Attention_18Grouped Query Attn1 × 1049028 × 4096
118Add_18_attnAdd1 × 1049028 × 4096
119RMSNorm_18_2RMSNorm1 × 1049028 × 4096
120MoE_18MoE Layer1 × 1049028 × 4096
121Add_18_ffnAdd1 × 1049028 × 4096
122RMSNorm_19_1RMSNorm1 × 1049028 × 4096
123Attention_19Grouped Query Attn1 × 1049028 × 4096
124Add_19_attnAdd1 × 1049028 × 4096
125RMSNorm_19_2RMSNorm1 × 1049028 × 4096
126MoE_19MoE Layer1 × 1049028 × 4096
127Add_19_ffnAdd1 × 1049028 × 4096
128RMSNorm_20_1RMSNorm1 × 1049028 × 4096
129Attention_20Grouped Query Attn1 × 1049028 × 4096
130Add_20_attnAdd1 × 1049028 × 4096
131RMSNorm_20_2RMSNorm1 × 1049028 × 4096
132MoE_20MoE Layer1 × 1049028 × 4096
133Add_20_ffnAdd1 × 1049028 × 4096
134RMSNorm_21_1RMSNorm1 × 1049028 × 4096
135Attention_21Grouped Query Attn1 × 1049028 × 4096
136Add_21_attnAdd1 × 1049028 × 4096
137RMSNorm_21_2RMSNorm1 × 1049028 × 4096
138MoE_21MoE Layer1 × 1049028 × 4096
139Add_21_ffnAdd1 × 1049028 × 4096
140RMSNorm_22_1RMSNorm1 × 1049028 × 4096
141Attention_22Grouped Query Attn1 × 1049028 × 4096
142Add_22_attnAdd1 × 1049028 × 4096
143RMSNorm_22_2RMSNorm1 × 1049028 × 4096
144MoE_22MoE Layer1 × 1049028 × 4096
145Add_22_ffnAdd1 × 1049028 × 4096
146RMSNorm_23_1RMSNorm1 × 1049028 × 4096
147Attention_23Grouped Query Attn1 × 1049028 × 4096
148Add_23_attnAdd1 × 1049028 × 4096
149RMSNorm_23_2RMSNorm1 × 1049028 × 4096
150MoE_23MoE Layer1 × 1049028 × 4096
151Add_23_ffnAdd1 × 1049028 × 4096
152RMSNorm_24_1RMSNorm1 × 1049028 × 4096
153Attention_24Grouped Query Attn1 × 1049028 × 4096
154Add_24_attnAdd1 × 1049028 × 4096
155RMSNorm_24_2RMSNorm1 × 1049028 × 4096
156MoE_24MoE Layer1 × 1049028 × 4096
157Add_24_ffnAdd1 × 1049028 × 4096
158RMSNorm_25_1RMSNorm1 × 1049028 × 4096
159Attention_25Grouped Query Attn1 × 1049028 × 4096
160Add_25_attnAdd1 × 1049028 × 4096
161RMSNorm_25_2RMSNorm1 × 1049028 × 4096
162MoE_25MoE Layer1 × 1049028 × 4096
163Add_25_ffnAdd1 × 1049028 × 4096
164RMSNorm_26_1RMSNorm1 × 1049028 × 4096
165Attention_26Grouped Query Attn1 × 1049028 × 4096
166Add_26_attnAdd1 × 1049028 × 4096
167RMSNorm_26_2RMSNorm1 × 1049028 × 4096
168MoE_26MoE Layer1 × 1049028 × 4096
169Add_26_ffnAdd1 × 1049028 × 4096
170RMSNorm_27_1RMSNorm1 × 1049028 × 4096
171Attention_27Grouped Query Attn1 × 1049028 × 4096
172Add_27_attnAdd1 × 1049028 × 4096
173RMSNorm_27_2RMSNorm1 × 1049028 × 4096
174MoE_27MoE Layer1 × 1049028 × 4096
175Add_27_ffnAdd1 × 1049028 × 4096
176RMSNorm_28_1RMSNorm1 × 1049028 × 4096
177Attention_28Grouped Query Attn1 × 1049028 × 4096
178Add_28_attnAdd1 × 1049028 × 4096
179RMSNorm_28_2RMSNorm1 × 1049028 × 4096
180MoE_28MoE Layer1 × 1049028 × 4096
181Add_28_ffnAdd1 × 1049028 × 4096
182RMSNorm_29_1RMSNorm1 × 1049028 × 4096
183Attention_29Grouped Query Attn1 × 1049028 × 4096
184Add_29_attnAdd1 × 1049028 × 4096
185RMSNorm_29_2RMSNorm1 × 1049028 × 4096
186MoE_29MoE Layer1 × 1049028 × 4096
187Add_29_ffnAdd1 × 1049028 × 4096
188RMSNorm_30_1RMSNorm1 × 1049028 × 4096
189Attention_30Grouped Query Attn1 × 1049028 × 4096
190Add_30_attnAdd1 × 1049028 × 4096
191RMSNorm_30_2RMSNorm1 × 1049028 × 4096
192MoE_30MoE Layer1 × 1049028 × 4096
193Add_30_ffnAdd1 × 1049028 × 4096
194RMSNorm_31_1RMSNorm1 × 1049028 × 4096
195Attention_31Grouped Query Attn1 × 1049028 × 4096
196Add_31_attnAdd1 × 1049028 × 4096
197RMSNorm_31_2RMSNorm1 × 1049028 × 4096
198MoE_31MoE Layer1 × 1049028 × 4096
199Add_31_ffnAdd1 × 1049028 × 4096
200RMSNorm_32_1RMSNorm1 × 1049028 × 4096
201Attention_32Grouped Query Attn1 × 1049028 × 4096
202Add_32_attnAdd1 × 1049028 × 4096
203RMSNorm_32_2RMSNorm1 × 1049028 × 4096
204MoE_32MoE Layer1 × 1049028 × 4096
205Add_32_ffnAdd1 × 1049028 × 4096
206RMSNorm_33_1RMSNorm1 × 1049028 × 4096
207Attention_33Grouped Query Attn1 × 1049028 × 4096
208Add_33_attnAdd1 × 1049028 × 4096
209RMSNorm_33_2RMSNorm1 × 1049028 × 4096
210MoE_33MoE Layer1 × 1049028 × 4096
211Add_33_ffnAdd1 × 1049028 × 4096
212RMSNorm_34_1RMSNorm1 × 1049028 × 4096
213Attention_34Grouped Query Attn1 × 1049028 × 4096
214Add_34_attnAdd1 × 1049028 × 4096
215RMSNorm_34_2RMSNorm1 × 1049028 × 4096
216MoE_34MoE Layer1 × 1049028 × 4096
217Add_34_ffnAdd1 × 1049028 × 4096
218RMSNorm_35_1RMSNorm1 × 1049028 × 4096
219Attention_35Grouped Query Attn1 × 1049028 × 4096
220Add_35_attnAdd1 × 1049028 × 4096
221RMSNorm_35_2RMSNorm1 × 1049028 × 4096
222MoE_35MoE Layer1 × 1049028 × 4096
223Add_35_ffnAdd1 × 1049028 × 4096
224RMSNorm_36_1RMSNorm1 × 1049028 × 4096
225Attention_36Grouped Query Attn1 × 1049028 × 4096
226Add_36_attnAdd1 × 1049028 × 4096
227RMSNorm_36_2RMSNorm1 × 1049028 × 4096
228MoE_36MoE Layer1 × 1049028 × 4096
229Add_36_ffnAdd1 × 1049028 × 4096
230RMSNorm_37_1RMSNorm1 × 1049028 × 4096
231Attention_37Grouped Query Attn1 × 1049028 × 4096
232Add_37_attnAdd1 × 1049028 × 4096
233RMSNorm_37_2RMSNorm1 × 1049028 × 4096
234MoE_37MoE Layer1 × 1049028 × 4096
235Add_37_ffnAdd1 × 1049028 × 4096
236RMSNorm_38_1RMSNorm1 × 1049028 × 4096
237Attention_38Grouped Query Attn1 × 1049028 × 4096
238Add_38_attnAdd1 × 1049028 × 4096
239RMSNorm_38_2RMSNorm1 × 1049028 × 4096
240MoE_38MoE Layer1 × 1049028 × 4096
241Add_38_ffnAdd1 × 1049028 × 4096
242RMSNorm_39_1RMSNorm1 × 1049028 × 4096
243Attention_39Grouped Query Attn1 × 1049028 × 4096
244Add_39_attnAdd1 × 1049028 × 4096
245RMSNorm_39_2RMSNorm1 × 1049028 × 4096
246MoE_39MoE Layer1 × 1049028 × 4096
247Add_39_ffnAdd1 × 1049028 × 4096
248RMSNorm_40_1RMSNorm1 × 1049028 × 4096
249Attention_40Grouped Query Attn1 × 1049028 × 4096
250Add_40_attnAdd1 × 1049028 × 4096
251RMSNorm_40_2RMSNorm1 × 1049028 × 4096
252MoE_40MoE Layer1 × 1049028 × 4096
253Add_40_ffnAdd1 × 1049028 × 4096
254RMSNorm_41_1RMSNorm1 × 1049028 × 4096
255Attention_41Grouped Query Attn1 × 1049028 × 4096
256Add_41_attnAdd1 × 1049028 × 4096
257RMSNorm_41_2RMSNorm1 × 1049028 × 4096
258MoE_41MoE Layer1 × 1049028 × 4096
259Add_41_ffnAdd1 × 1049028 × 4096
260RMSNorm_42_1RMSNorm1 × 1049028 × 4096
261Attention_42Grouped Query Attn1 × 1049028 × 4096
262Add_42_attnAdd1 × 1049028 × 4096
263RMSNorm_42_2RMSNorm1 × 1049028 × 4096
264MoE_42MoE Layer1 × 1049028 × 4096
265Add_42_ffnAdd1 × 1049028 × 4096
266RMSNorm_43_1RMSNorm1 × 1049028 × 4096
267Attention_43Grouped Query Attn1 × 1049028 × 4096
268Add_43_attnAdd1 × 1049028 × 4096
269RMSNorm_43_2RMSNorm1 × 1049028 × 4096
270MoE_43MoE Layer1 × 1049028 × 4096
271Add_43_ffnAdd1 × 1049028 × 4096
272RMSNorm_44_1RMSNorm1 × 1049028 × 4096
273Attention_44Grouped Query Attn1 × 1049028 × 4096
274Add_44_attnAdd1 × 1049028 × 4096
275RMSNorm_44_2RMSNorm1 × 1049028 × 4096
276MoE_44MoE Layer1 × 1049028 × 4096
277Add_44_ffnAdd1 × 1049028 × 4096
278RMSNorm_45_1RMSNorm1 × 1049028 × 4096
279Attention_45Grouped Query Attn1 × 1049028 × 4096
280Add_45_attnAdd1 × 1049028 × 4096
281RMSNorm_45_2RMSNorm1 × 1049028 × 4096
282MoE_45MoE Layer1 × 1049028 × 4096
283Add_45_ffnAdd1 × 1049028 × 4096
284RMSNorm_46_1RMSNorm1 × 1049028 × 4096
285Attention_46Grouped Query Attn1 × 1049028 × 4096
286Add_46_attnAdd1 × 1049028 × 4096
287RMSNorm_46_2RMSNorm1 × 1049028 × 4096
288MoE_46MoE Layer1 × 1049028 × 4096
289Add_46_ffnAdd1 × 1049028 × 4096
290RMSNorm_47_1RMSNorm1 × 1049028 × 4096
291Attention_47Grouped Query Attn1 × 1049028 × 4096
292Add_47_attnAdd1 × 1049028 × 4096
293RMSNorm_47_2RMSNorm1 × 1049028 × 4096
294MoE_47MoE Layer1 × 1049028 × 4096
295Add_47_ffnAdd1 × 1049028 × 4096
296RMSNorm_48_1RMSNorm1 × 1049028 × 4096
297Attention_48Grouped Query Attn1 × 1049028 × 4096
298Add_48_attnAdd1 × 1049028 × 4096
299RMSNorm_48_2RMSNorm1 × 1049028 × 4096
300MoE_48MoE Layer1 × 1049028 × 4096
301Add_48_ffnAdd1 × 1049028 × 4096
302Final_RMSNormRMSNorm1 × 1049028 × 4096
303LM_HeadLinear1 × 1049028 × 152576
304OutputOutput1 × 1049028 × 152576

What the verifier says

warn"RoPE" receives input but its output is not connected. This layer will be unreachable in the forward pass. Fix: Connect the output forward, or add an Output node if this is the final layer.
dead-end
infoMoE layers require an auxiliary router z-loss + load-balance loss during training to prevent expert collapse. This is not visible in the architecture diagram but must be in the training loop. Applies to all 48: MoE_1, MoE_2, MoE_3, MoE_4, MoE_5, MoE_6, MoE_7, MoE_8, MoE_9, MoE_10, MoE_11, MoE_12, MoE_13, MoE_14, MoE_15, MoE_16, MoE_17, MoE_18, MoE_19, MoE_20, MoE_21, MoE_22, MoE_23, MoE_24, MoE_25, MoE_26, MoE_27, MoE_28, MoE_29, MoE_30, MoE_31, MoE_32, MoE_33, MoE_34, MoE_35, MoE_36, MoE_37, MoE_38, MoE_39, MoE_40, MoE_41, MoE_42, MoE_43, MoE_44, MoE_45, MoE_46, MoE_47, MoE_48. Fix: Add a note on these layers. Typical aux_loss coefficient: 1e-2 (Mixtral/Switch Transformer).
moe-no-aux-loss
infoAt 48 stacked attention layers, residual-branch outputs add up; unscaled init lets activation variance grow with depth. GPT-2/LLaMA-family models scale the residual projections by depth (N(0, 0.02 / √(2L))). Fix: Scale residual output projections by depth: nn.init.normal_(w, std=0.02 / math.sqrt(2 * n_layers))
deep-attention-default-init

Do this to your own model

Same numbers, on a model in your repo, in one command. No account.

pip install neurarch-trace
neurarch-trace XiaomiMiMo/MiMo-V2.5 --plan --share