Latent decoder

Pinned implementation and paper LDLM @ e18c42a, not yet public · arXiv:2605.07933v1 · OpenWebText · 1024 · checkpoint, tokenizer and saved statistics not verified · How to read

Latent decoder

Positions to queries

×

no parameters

Add each position’s learned vector to its query, from the same table as the latents’.

r0=e~+Ep{\htmlData{light=start add-query-positions}{r_0}}={\htmlData{light=query-scale}{\tilde e}}+{\htmlData{light=positions}{E_{p}}}
in
e~\tilde e [B batch, T positions, 768 d]EpE_{p} [B batch, T positions, 768 d]
out
r0r_0 [B batch, T positions, 768 d]
Sources1

Direct evidence

Every query is the mask vector: it passes the MLP and a scale, positions are added; the latents take the same position table; the blocks are applied; every position counts as masked.

The implementation, pinned at revision e18c42acdfb7, not yet public