Token decoder › 3 blocks › Self-attention › Attention over all positions
Softmax over keys
×no parameters
Turn each query’s scores over all positions into weights that sum to one.
- in
- [B batch, 12 heads, T queries, T keys]
- out
- [B batch, 12 heads, T queries, T keys]
No position is masked, and the attention dropout is 0.
Sources1
Direct evidence
A block normalizes, modulates by the time condition’s shift and scale, projects to queries, keys and values, rotates queries and keys, attends over all positions with softcap, and adds the projected output scaled by a gate; the MLP branch does the same with its own shift, scale and gate.
The implementation, pinned at revision e18c42acdfb7, not yet public