N Neurarch Architectures Models Checks Data Docs Open the app

Models / modernbert

gte-reranker-modernbert-base

Reconstructed from its own config.json with no weights read. 2.7M 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
136M
135,950,592 parameters
In the published checkpoint
150M
149,605,633 scalars · safetensors.total, read 2025-07-04
Delta
-9.13%

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

Structure

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

LayerTypeOutput shape
1InputInput1 × 8192
2EmbeddingEmbedding1 × 8192 × 768
3Positional_EmbeddingLearned Pos Embed1 × 8192 × 768
4Attention_1Multi-Head Attention1 × 8192 × 768
5Add_1Add1 × 8192 × 768
6LayerNorm_1_1LayerNorm1 × 8192 × 768
7FFN_1Feed Forward1 × 8192 × 768
8Attention_2Multi-Head Attention1 × 8192 × 768
9Add_2Add1 × 8192 × 768
10LayerNorm_2_1LayerNorm1 × 8192 × 768
11FFN_2Feed Forward1 × 8192 × 768
12Attention_3Multi-Head Attention1 × 8192 × 768
13Add_3Add1 × 8192 × 768
14LayerNorm_3_1LayerNorm1 × 8192 × 768
15FFN_3Feed Forward1 × 8192 × 768
16Attention_4Multi-Head Attention1 × 8192 × 768
17Add_4Add1 × 8192 × 768
18LayerNorm_4_1LayerNorm1 × 8192 × 768
19FFN_4Feed Forward1 × 8192 × 768
20Attention_5Multi-Head Attention1 × 8192 × 768
21Add_5Add1 × 8192 × 768
22LayerNorm_5_1LayerNorm1 × 8192 × 768
23FFN_5Feed Forward1 × 8192 × 768
24Attention_6Multi-Head Attention1 × 8192 × 768
25Add_6Add1 × 8192 × 768
26LayerNorm_6_1LayerNorm1 × 8192 × 768
27FFN_6Feed Forward1 × 8192 × 768
28Attention_7Multi-Head Attention1 × 8192 × 768
29Add_7Add1 × 8192 × 768
30LayerNorm_7_1LayerNorm1 × 8192 × 768
31FFN_7Feed Forward1 × 8192 × 768
32Attention_8Multi-Head Attention1 × 8192 × 768
33Add_8Add1 × 8192 × 768
34LayerNorm_8_1LayerNorm1 × 8192 × 768
35FFN_8Feed Forward1 × 8192 × 768
36Attention_9Multi-Head Attention1 × 8192 × 768
37Add_9Add1 × 8192 × 768
38LayerNorm_9_1LayerNorm1 × 8192 × 768
39FFN_9Feed Forward1 × 8192 × 768
40Attention_10Multi-Head Attention1 × 8192 × 768
41Add_10Add1 × 8192 × 768
42LayerNorm_10_1LayerNorm1 × 8192 × 768
43FFN_10Feed Forward1 × 8192 × 768
44Attention_11Multi-Head Attention1 × 8192 × 768
45Add_11Add1 × 8192 × 768
46LayerNorm_11_1LayerNorm1 × 8192 × 768
47FFN_11Feed Forward1 × 8192 × 768
48Attention_12Multi-Head Attention1 × 8192 × 768
49Add_12Add1 × 8192 × 768
50LayerNorm_12_1LayerNorm1 × 8192 × 768
51FFN_12Feed Forward1 × 8192 × 768
52Attention_13Multi-Head Attention1 × 8192 × 768
53Add_13Add1 × 8192 × 768
54LayerNorm_13_1LayerNorm1 × 8192 × 768
55FFN_13Feed Forward1 × 8192 × 768
56Attention_14Multi-Head Attention1 × 8192 × 768
57Add_14Add1 × 8192 × 768
58LayerNorm_14_1LayerNorm1 × 8192 × 768
59FFN_14Feed Forward1 × 8192 × 768
60Attention_15Multi-Head Attention1 × 8192 × 768
61Add_15Add1 × 8192 × 768
62LayerNorm_15_1LayerNorm1 × 8192 × 768
63FFN_15Feed Forward1 × 8192 × 768
64Attention_16Multi-Head Attention1 × 8192 × 768
65Add_16Add1 × 8192 × 768
66LayerNorm_16_1LayerNorm1 × 8192 × 768
67FFN_16Feed Forward1 × 8192 × 768
68Attention_17Multi-Head Attention1 × 8192 × 768
69Add_17Add1 × 8192 × 768
70LayerNorm_17_1LayerNorm1 × 8192 × 768
71FFN_17Feed Forward1 × 8192 × 768
72Attention_18Multi-Head Attention1 × 8192 × 768
73Add_18Add1 × 8192 × 768
74LayerNorm_18_1LayerNorm1 × 8192 × 768
75FFN_18Feed Forward1 × 8192 × 768
76Attention_19Multi-Head Attention1 × 8192 × 768
77Add_19Add1 × 8192 × 768
78LayerNorm_19_1LayerNorm1 × 8192 × 768
79FFN_19Feed Forward1 × 8192 × 768
80Attention_20Multi-Head Attention1 × 8192 × 768
81Add_20Add1 × 8192 × 768
82LayerNorm_20_1LayerNorm1 × 8192 × 768
83FFN_20Feed Forward1 × 8192 × 768
84Attention_21Multi-Head Attention1 × 8192 × 768
85Add_21Add1 × 8192 × 768
86LayerNorm_21_1LayerNorm1 × 8192 × 768
87FFN_21Feed Forward1 × 8192 × 768
88Attention_22Multi-Head Attention1 × 8192 × 768
89Add_22Add1 × 8192 × 768
90LayerNorm_22_1LayerNorm1 × 8192 × 768
91FFN_22Feed Forward1 × 8192 × 768
92OutputOutput1 × 8192 × 768

What the verifier says

infoAt 22 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 Alibaba-NLP/gte-reranker-modernbert-base --plan --share

Other modernbert checkpoints

granite-embedding-small-english-r2
43.8M derived · -8.16% against the checkpoint
ModernBERT-base
136M derived · -9.16% against the checkpoint