N Neurarch Architectures Models Checks Data Docs Open the app

Models / llama

Llama-3.2-1B-Instruct

Reconstructed from its own config.json with no weights read. 6.1M 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
1.24B
1,235,812,352 parameters
In the published checkpoint
1.24B
1,235,814,400 scalars · safetensors.total, read 2024-10-24
Delta
-0.00%

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
98
Will it forward-pass
Yes
Priced on
A10G (24GB)
Est. one run
$25844.38
CardMemory
T4 (16GB)weights + activationsdoes not fit
A100 (40GB)weights + activationsfits
H100 (80GB)weights + activationsfits

Structure

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

LayerTypeOutput shape
1InputInput1 × 131072
2EmbeddingEmbedding1 × 131072 × 2048
3RoPERoPE1 × 131072 × 2048
4RMSNorm_1_1RMSNorm1 × 131072 × 2048
5Attention_1Grouped Query Attn1 × 131072 × 2048
6Add_1_attnAdd1 × 131072 × 2048
7RMSNorm_1_2RMSNorm1 × 131072 × 2048
8FFN_1SwiGLU1 × 131072 × 2048
9Add_1_ffnAdd1 × 131072 × 2048
10RMSNorm_2_1RMSNorm1 × 131072 × 2048
11Attention_2Grouped Query Attn1 × 131072 × 2048
12Add_2_attnAdd1 × 131072 × 2048
13RMSNorm_2_2RMSNorm1 × 131072 × 2048
14FFN_2SwiGLU1 × 131072 × 2048
15Add_2_ffnAdd1 × 131072 × 2048
16RMSNorm_3_1RMSNorm1 × 131072 × 2048
17Attention_3Grouped Query Attn1 × 131072 × 2048
18Add_3_attnAdd1 × 131072 × 2048
19RMSNorm_3_2RMSNorm1 × 131072 × 2048
20FFN_3SwiGLU1 × 131072 × 2048
21Add_3_ffnAdd1 × 131072 × 2048
22RMSNorm_4_1RMSNorm1 × 131072 × 2048
23Attention_4Grouped Query Attn1 × 131072 × 2048
24Add_4_attnAdd1 × 131072 × 2048
25RMSNorm_4_2RMSNorm1 × 131072 × 2048
26FFN_4SwiGLU1 × 131072 × 2048
27Add_4_ffnAdd1 × 131072 × 2048
28RMSNorm_5_1RMSNorm1 × 131072 × 2048
29Attention_5Grouped Query Attn1 × 131072 × 2048
30Add_5_attnAdd1 × 131072 × 2048
31RMSNorm_5_2RMSNorm1 × 131072 × 2048
32FFN_5SwiGLU1 × 131072 × 2048
33Add_5_ffnAdd1 × 131072 × 2048
34RMSNorm_6_1RMSNorm1 × 131072 × 2048
35Attention_6Grouped Query Attn1 × 131072 × 2048
36Add_6_attnAdd1 × 131072 × 2048
37RMSNorm_6_2RMSNorm1 × 131072 × 2048
38FFN_6SwiGLU1 × 131072 × 2048
39Add_6_ffnAdd1 × 131072 × 2048
40RMSNorm_7_1RMSNorm1 × 131072 × 2048
41Attention_7Grouped Query Attn1 × 131072 × 2048
42Add_7_attnAdd1 × 131072 × 2048
43RMSNorm_7_2RMSNorm1 × 131072 × 2048
44FFN_7SwiGLU1 × 131072 × 2048
45Add_7_ffnAdd1 × 131072 × 2048
46RMSNorm_8_1RMSNorm1 × 131072 × 2048
47Attention_8Grouped Query Attn1 × 131072 × 2048
48Add_8_attnAdd1 × 131072 × 2048
49RMSNorm_8_2RMSNorm1 × 131072 × 2048
50FFN_8SwiGLU1 × 131072 × 2048
51Add_8_ffnAdd1 × 131072 × 2048
52RMSNorm_9_1RMSNorm1 × 131072 × 2048
53Attention_9Grouped Query Attn1 × 131072 × 2048
54Add_9_attnAdd1 × 131072 × 2048
55RMSNorm_9_2RMSNorm1 × 131072 × 2048
56FFN_9SwiGLU1 × 131072 × 2048
57Add_9_ffnAdd1 × 131072 × 2048
58RMSNorm_10_1RMSNorm1 × 131072 × 2048
59Attention_10Grouped Query Attn1 × 131072 × 2048
60Add_10_attnAdd1 × 131072 × 2048
61RMSNorm_10_2RMSNorm1 × 131072 × 2048
62FFN_10SwiGLU1 × 131072 × 2048
63Add_10_ffnAdd1 × 131072 × 2048
64RMSNorm_11_1RMSNorm1 × 131072 × 2048
65Attention_11Grouped Query Attn1 × 131072 × 2048
66Add_11_attnAdd1 × 131072 × 2048
67RMSNorm_11_2RMSNorm1 × 131072 × 2048
68FFN_11SwiGLU1 × 131072 × 2048
69Add_11_ffnAdd1 × 131072 × 2048
70RMSNorm_12_1RMSNorm1 × 131072 × 2048
71Attention_12Grouped Query Attn1 × 131072 × 2048
72Add_12_attnAdd1 × 131072 × 2048
73RMSNorm_12_2RMSNorm1 × 131072 × 2048
74FFN_12SwiGLU1 × 131072 × 2048
75Add_12_ffnAdd1 × 131072 × 2048
76RMSNorm_13_1RMSNorm1 × 131072 × 2048
77Attention_13Grouped Query Attn1 × 131072 × 2048
78Add_13_attnAdd1 × 131072 × 2048
79RMSNorm_13_2RMSNorm1 × 131072 × 2048
80FFN_13SwiGLU1 × 131072 × 2048
81Add_13_ffnAdd1 × 131072 × 2048
82RMSNorm_14_1RMSNorm1 × 131072 × 2048
83Attention_14Grouped Query Attn1 × 131072 × 2048
84Add_14_attnAdd1 × 131072 × 2048
85RMSNorm_14_2RMSNorm1 × 131072 × 2048
86FFN_14SwiGLU1 × 131072 × 2048
87Add_14_ffnAdd1 × 131072 × 2048
88RMSNorm_15_1RMSNorm1 × 131072 × 2048
89Attention_15Grouped Query Attn1 × 131072 × 2048
90Add_15_attnAdd1 × 131072 × 2048
91RMSNorm_15_2RMSNorm1 × 131072 × 2048
92FFN_15SwiGLU1 × 131072 × 2048
93Add_15_ffnAdd1 × 131072 × 2048
94RMSNorm_16_1RMSNorm1 × 131072 × 2048
95Attention_16Grouped Query Attn1 × 131072 × 2048
96Add_16_attnAdd1 × 131072 × 2048
97RMSNorm_16_2RMSNorm1 × 131072 × 2048
98FFN_16SwiGLU1 × 131072 × 2048
99Add_16_ffnAdd1 × 131072 × 2048
100OutputOutput1 × 131072 × 2048

What the verifier says

infoAt 16 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 meta-llama/Llama-3.2-1B-Instruct --plan --share

Other llama checkpoints

dolphin-2.9.1-yi-1.5-34b
34.39B derived · +0.00% against the checkpoint
Llama-3.1-8B-Instruct
8.03B derived · +0.00% against the checkpoint
llama2-embedding-1b-8k
1.07B derived · +13.2% against the checkpoint
Meta-Llama-3.1-8B-Instruct
8.03B derived · +0.00% against the checkpoint