jina-embeddings-v4

Reconstructed from its own config.json with no weights read. 357K 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

3.09B

3,089,550,848 parameters

In the published checkpoint

3.75B

3,754,885,248 scalars · safetensors.total, read 2026-04-08

Delta

-17.7%

custom-code This repository ships its own modeling code (`auto_map`, e.g. `configuration_jina_embeddings_v4.py`), so `config.json` names a class in the repo rather than an architecture `transformers` defines. The graph below is what those config keys mean under `transformers` semantics, which is not necessarily what the repo's own file builds. A gap here is a statement about what we read, not about the checkpoint.

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

223

Will it forward-pass

Yes

Priced on

A10G (24GB)

Est. one run

$58713.01

Card	Memory
T4 (16GB)	weights + activations	does not fit
A100 (40GB)	weights + activations	does not fit
H100 (80GB)	weights + activations	fits

Structure

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

	Layer	Type	Output shape
1	Input	Input	1 × 128000
2	Embedding	Embedding	1 × 128000 × 2048
3	RoPE	RoPE	1 × 128000 × 2048
4	Vision input	Input	3 × 224 × 224
5	PatchEmbed	Patch Embed	256 × 1280
6	Patch_Position_Embedding	Learned Pos Embed	256 × 1280
7	Vision encoder (internals not in config)	Projection	256 × 2048
8	Vision tokens	Reshape	1 × 256 × 2048
9	Multimodal fusion (concat tokens)	Concatenate	1 × 128256 × 2048
10	RMSNorm_1_1	RMSNorm	1 × 128256 × 2048
11	Attention_1	Grouped Query Attn	1 × 128256 × 2048
12	Add_1_attn	Add	1 × 128256 × 2048
13	RMSNorm_1_2	RMSNorm	1 × 128256 × 2048
14	FFN_1	SwiGLU	1 × 128256 × 2048
15	Add_1_ffn	Add	1 × 128256 × 2048
16	RMSNorm_2_1	RMSNorm	1 × 128256 × 2048
17	Attention_2	Grouped Query Attn	1 × 128256 × 2048
18	Add_2_attn	Add	1 × 128256 × 2048
19	RMSNorm_2_2	RMSNorm	1 × 128256 × 2048
20	FFN_2	SwiGLU	1 × 128256 × 2048
21	Add_2_ffn	Add	1 × 128256 × 2048
22	RMSNorm_3_1	RMSNorm	1 × 128256 × 2048
23	Attention_3	Grouped Query Attn	1 × 128256 × 2048
24	Add_3_attn	Add	1 × 128256 × 2048
25	RMSNorm_3_2	RMSNorm	1 × 128256 × 2048
26	FFN_3	SwiGLU	1 × 128256 × 2048
27	Add_3_ffn	Add	1 × 128256 × 2048
28	RMSNorm_4_1	RMSNorm	1 × 128256 × 2048
29	Attention_4	Grouped Query Attn	1 × 128256 × 2048
30	Add_4_attn	Add	1 × 128256 × 2048
31	RMSNorm_4_2	RMSNorm	1 × 128256 × 2048
32	FFN_4	SwiGLU	1 × 128256 × 2048
33	Add_4_ffn	Add	1 × 128256 × 2048
34	RMSNorm_5_1	RMSNorm	1 × 128256 × 2048
35	Attention_5	Grouped Query Attn	1 × 128256 × 2048
36	Add_5_attn	Add	1 × 128256 × 2048
37	RMSNorm_5_2	RMSNorm	1 × 128256 × 2048
38	FFN_5	SwiGLU	1 × 128256 × 2048
39	Add_5_ffn	Add	1 × 128256 × 2048
40	RMSNorm_6_1	RMSNorm	1 × 128256 × 2048
41	Attention_6	Grouped Query Attn	1 × 128256 × 2048
42	Add_6_attn	Add	1 × 128256 × 2048
43	RMSNorm_6_2	RMSNorm	1 × 128256 × 2048
44	FFN_6	SwiGLU	1 × 128256 × 2048
45	Add_6_ffn	Add	1 × 128256 × 2048
46	RMSNorm_7_1	RMSNorm	1 × 128256 × 2048
47	Attention_7	Grouped Query Attn	1 × 128256 × 2048
48	Add_7_attn	Add	1 × 128256 × 2048
49	RMSNorm_7_2	RMSNorm	1 × 128256 × 2048
50	FFN_7	SwiGLU	1 × 128256 × 2048
51	Add_7_ffn	Add	1 × 128256 × 2048
52	RMSNorm_8_1	RMSNorm	1 × 128256 × 2048
53	Attention_8	Grouped Query Attn	1 × 128256 × 2048
54	Add_8_attn	Add	1 × 128256 × 2048
55	RMSNorm_8_2	RMSNorm	1 × 128256 × 2048
56	FFN_8	SwiGLU	1 × 128256 × 2048
57	Add_8_ffn	Add	1 × 128256 × 2048
58	RMSNorm_9_1	RMSNorm	1 × 128256 × 2048
59	Attention_9	Grouped Query Attn	1 × 128256 × 2048
60	Add_9_attn	Add	1 × 128256 × 2048
61	RMSNorm_9_2	RMSNorm	1 × 128256 × 2048
62	FFN_9	SwiGLU	1 × 128256 × 2048
63	Add_9_ffn	Add	1 × 128256 × 2048
64	RMSNorm_10_1	RMSNorm	1 × 128256 × 2048
65	Attention_10	Grouped Query Attn	1 × 128256 × 2048
66	Add_10_attn	Add	1 × 128256 × 2048
67	RMSNorm_10_2	RMSNorm	1 × 128256 × 2048
68	FFN_10	SwiGLU	1 × 128256 × 2048
69	Add_10_ffn	Add	1 × 128256 × 2048
70	RMSNorm_11_1	RMSNorm	1 × 128256 × 2048
71	Attention_11	Grouped Query Attn	1 × 128256 × 2048
72	Add_11_attn	Add	1 × 128256 × 2048
73	RMSNorm_11_2	RMSNorm	1 × 128256 × 2048
74	FFN_11	SwiGLU	1 × 128256 × 2048
75	Add_11_ffn	Add	1 × 128256 × 2048
76	RMSNorm_12_1	RMSNorm	1 × 128256 × 2048
77	Attention_12	Grouped Query Attn	1 × 128256 × 2048
78	Add_12_attn	Add	1 × 128256 × 2048
79	RMSNorm_12_2	RMSNorm	1 × 128256 × 2048
80	FFN_12	SwiGLU	1 × 128256 × 2048
81	Add_12_ffn	Add	1 × 128256 × 2048
82	RMSNorm_13_1	RMSNorm	1 × 128256 × 2048
83	Attention_13	Grouped Query Attn	1 × 128256 × 2048
84	Add_13_attn	Add	1 × 128256 × 2048
85	RMSNorm_13_2	RMSNorm	1 × 128256 × 2048
86	FFN_13	SwiGLU	1 × 128256 × 2048
87	Add_13_ffn	Add	1 × 128256 × 2048
88	RMSNorm_14_1	RMSNorm	1 × 128256 × 2048
89	Attention_14	Grouped Query Attn	1 × 128256 × 2048
90	Add_14_attn	Add	1 × 128256 × 2048
91	RMSNorm_14_2	RMSNorm	1 × 128256 × 2048
92	FFN_14	SwiGLU	1 × 128256 × 2048
93	Add_14_ffn	Add	1 × 128256 × 2048
94	RMSNorm_15_1	RMSNorm	1 × 128256 × 2048
95	Attention_15	Grouped Query Attn	1 × 128256 × 2048
96	Add_15_attn	Add	1 × 128256 × 2048
97	RMSNorm_15_2	RMSNorm	1 × 128256 × 2048
98	FFN_15	SwiGLU	1 × 128256 × 2048
99	Add_15_ffn	Add	1 × 128256 × 2048
100	RMSNorm_16_1	RMSNorm	1 × 128256 × 2048
101	Attention_16	Grouped Query Attn	1 × 128256 × 2048
102	Add_16_attn	Add	1 × 128256 × 2048
103	RMSNorm_16_2	RMSNorm	1 × 128256 × 2048
104	FFN_16	SwiGLU	1 × 128256 × 2048
105	Add_16_ffn	Add	1 × 128256 × 2048
106	RMSNorm_17_1	RMSNorm	1 × 128256 × 2048
107	Attention_17	Grouped Query Attn	1 × 128256 × 2048
108	Add_17_attn	Add	1 × 128256 × 2048
109	RMSNorm_17_2	RMSNorm	1 × 128256 × 2048
110	FFN_17	SwiGLU	1 × 128256 × 2048
111	Add_17_ffn	Add	1 × 128256 × 2048
112	RMSNorm_18_1	RMSNorm	1 × 128256 × 2048
113	Attention_18	Grouped Query Attn	1 × 128256 × 2048
114	Add_18_attn	Add	1 × 128256 × 2048
115	RMSNorm_18_2	RMSNorm	1 × 128256 × 2048
116	FFN_18	SwiGLU	1 × 128256 × 2048
117	Add_18_ffn	Add	1 × 128256 × 2048
118	RMSNorm_19_1	RMSNorm	1 × 128256 × 2048
119	Attention_19	Grouped Query Attn	1 × 128256 × 2048
120	Add_19_attn	Add	1 × 128256 × 2048
121	RMSNorm_19_2	RMSNorm	1 × 128256 × 2048
122	FFN_19	SwiGLU	1 × 128256 × 2048
123	Add_19_ffn	Add	1 × 128256 × 2048
124	RMSNorm_20_1	RMSNorm	1 × 128256 × 2048
125	Attention_20	Grouped Query Attn	1 × 128256 × 2048
126	Add_20_attn	Add	1 × 128256 × 2048
127	RMSNorm_20_2	RMSNorm	1 × 128256 × 2048
128	FFN_20	SwiGLU	1 × 128256 × 2048
129	Add_20_ffn	Add	1 × 128256 × 2048
130	RMSNorm_21_1	RMSNorm	1 × 128256 × 2048
131	Attention_21	Grouped Query Attn	1 × 128256 × 2048
132	Add_21_attn	Add	1 × 128256 × 2048
133	RMSNorm_21_2	RMSNorm	1 × 128256 × 2048
134	FFN_21	SwiGLU	1 × 128256 × 2048
135	Add_21_ffn	Add	1 × 128256 × 2048
136	RMSNorm_22_1	RMSNorm	1 × 128256 × 2048
137	Attention_22	Grouped Query Attn	1 × 128256 × 2048
138	Add_22_attn	Add	1 × 128256 × 2048
139	RMSNorm_22_2	RMSNorm	1 × 128256 × 2048
140	FFN_22	SwiGLU	1 × 128256 × 2048
141	Add_22_ffn	Add	1 × 128256 × 2048
142	RMSNorm_23_1	RMSNorm	1 × 128256 × 2048
143	Attention_23	Grouped Query Attn	1 × 128256 × 2048
144	Add_23_attn	Add	1 × 128256 × 2048
145	RMSNorm_23_2	RMSNorm	1 × 128256 × 2048
146	FFN_23	SwiGLU	1 × 128256 × 2048
147	Add_23_ffn	Add	1 × 128256 × 2048
148	RMSNorm_24_1	RMSNorm	1 × 128256 × 2048
149	Attention_24	Grouped Query Attn	1 × 128256 × 2048
150	Add_24_attn	Add	1 × 128256 × 2048
151	RMSNorm_24_2	RMSNorm	1 × 128256 × 2048
152	FFN_24	SwiGLU	1 × 128256 × 2048
153	Add_24_ffn	Add	1 × 128256 × 2048
154	RMSNorm_25_1	RMSNorm	1 × 128256 × 2048
155	Attention_25	Grouped Query Attn	1 × 128256 × 2048
156	Add_25_attn	Add	1 × 128256 × 2048
157	RMSNorm_25_2	RMSNorm	1 × 128256 × 2048
158	FFN_25	SwiGLU	1 × 128256 × 2048
159	Add_25_ffn	Add	1 × 128256 × 2048
160	RMSNorm_26_1	RMSNorm	1 × 128256 × 2048
161	Attention_26	Grouped Query Attn	1 × 128256 × 2048
162	Add_26_attn	Add	1 × 128256 × 2048
163	RMSNorm_26_2	RMSNorm	1 × 128256 × 2048
164	FFN_26	SwiGLU	1 × 128256 × 2048
165	Add_26_ffn	Add	1 × 128256 × 2048
166	RMSNorm_27_1	RMSNorm	1 × 128256 × 2048
167	Attention_27	Grouped Query Attn	1 × 128256 × 2048
168	Add_27_attn	Add	1 × 128256 × 2048
169	RMSNorm_27_2	RMSNorm	1 × 128256 × 2048
170	FFN_27	SwiGLU	1 × 128256 × 2048
171	Add_27_ffn	Add	1 × 128256 × 2048
172	RMSNorm_28_1	RMSNorm	1 × 128256 × 2048
173	Attention_28	Grouped Query Attn	1 × 128256 × 2048
174	Add_28_attn	Add	1 × 128256 × 2048
175	RMSNorm_28_2	RMSNorm	1 × 128256 × 2048
176	FFN_28	SwiGLU	1 × 128256 × 2048
177	Add_28_ffn	Add	1 × 128256 × 2048
178	RMSNorm_29_1	RMSNorm	1 × 128256 × 2048
179	Attention_29	Grouped Query Attn	1 × 128256 × 2048
180	Add_29_attn	Add	1 × 128256 × 2048
181	RMSNorm_29_2	RMSNorm	1 × 128256 × 2048
182	FFN_29	SwiGLU	1 × 128256 × 2048
183	Add_29_ffn	Add	1 × 128256 × 2048
184	RMSNorm_30_1	RMSNorm	1 × 128256 × 2048
185	Attention_30	Grouped Query Attn	1 × 128256 × 2048
186	Add_30_attn	Add	1 × 128256 × 2048
187	RMSNorm_30_2	RMSNorm	1 × 128256 × 2048
188	FFN_30	SwiGLU	1 × 128256 × 2048
189	Add_30_ffn	Add	1 × 128256 × 2048
190	RMSNorm_31_1	RMSNorm	1 × 128256 × 2048
191	Attention_31	Grouped Query Attn	1 × 128256 × 2048
192	Add_31_attn	Add	1 × 128256 × 2048
193	RMSNorm_31_2	RMSNorm	1 × 128256 × 2048
194	FFN_31	SwiGLU	1 × 128256 × 2048
195	Add_31_ffn	Add	1 × 128256 × 2048
196	RMSNorm_32_1	RMSNorm	1 × 128256 × 2048
197	Attention_32	Grouped Query Attn	1 × 128256 × 2048
198	Add_32_attn	Add	1 × 128256 × 2048
199	RMSNorm_32_2	RMSNorm	1 × 128256 × 2048
200	FFN_32	SwiGLU	1 × 128256 × 2048
201	Add_32_ffn	Add	1 × 128256 × 2048
202	RMSNorm_33_1	RMSNorm	1 × 128256 × 2048
203	Attention_33	Grouped Query Attn	1 × 128256 × 2048
204	Add_33_attn	Add	1 × 128256 × 2048
205	RMSNorm_33_2	RMSNorm	1 × 128256 × 2048
206	FFN_33	SwiGLU	1 × 128256 × 2048
207	Add_33_ffn	Add	1 × 128256 × 2048
208	RMSNorm_34_1	RMSNorm	1 × 128256 × 2048
209	Attention_34	Grouped Query Attn	1 × 128256 × 2048
210	Add_34_attn	Add	1 × 128256 × 2048
211	RMSNorm_34_2	RMSNorm	1 × 128256 × 2048
212	FFN_34	SwiGLU	1 × 128256 × 2048
213	Add_34_ffn	Add	1 × 128256 × 2048
214	RMSNorm_35_1	RMSNorm	1 × 128256 × 2048
215	Attention_35	Grouped Query Attn	1 × 128256 × 2048
216	Add_35_attn	Add	1 × 128256 × 2048
217	RMSNorm_35_2	RMSNorm	1 × 128256 × 2048
218	FFN_35	SwiGLU	1 × 128256 × 2048
219	Add_35_ffn	Add	1 × 128256 × 2048
220	RMSNorm_36_1	RMSNorm	1 × 128256 × 2048
221	Attention_36	Grouped Query Attn	1 × 128256 × 2048
222	Add_36_attn	Add	1 × 128256 × 2048
223	RMSNorm_36_2	RMSNorm	1 × 128256 × 2048
224	FFN_36	SwiGLU	1 × 128256 × 2048
225	Add_36_ffn	Add	1 × 128256 × 2048
226	Output	Output	1 × 128256 × 2048

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

infoLLaMA uses intermediateSize ≈ ⌊(8/3 × D) / 256⌋ × 256. Current: 11008 (5.38× embedDim). Expected: ~5376. Fix: Set intermediateSize to 5376 for embedDim=2048.
swiglu-dim-convention

infoAt 36 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 jinaai/jina-embeddings-v4 --plan --share

Paste a model instead Markdown twin Index JSON POST /api/v1/plan

Other checkpoints

jina-reranker-v2-base-multilingual

278M derived · -0.22% against the checkpoint