Token decoder › 3 blocks › Self-attention › Attention over all positions
Scale
×no parameters
Divide the scores by √64.
- in
- [B batch, 12 heads, T queries, T keys]
- out
- [B batch, 12 heads, T queries, T keys]
The scale 1/√64 is the default of flash-attn 2.7.4.post1 and runs inside its kernel, as the cap does (flash_api.cpp 108–121). The library is pinned by its source archive in the review record.
Sources2
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
Direct evidence
The token decoder’s blocks: no softcap, no time conditioning, the gated MLP with the configured multiplier.
The implementation, pinned at revision e18c42acdfb7, not yet public