Latent encoder › 6 encoder blocks › MLP
Contraction
×learned
Map 3,072 features back to 768.
- in
- [B batch, T positions, 3072 d_ff]
- out
- [B batch, T positions, 768 d]
- parameters
- 2,360,064 · × 6 blocks = 14,160,384
Parameters
2,360,064 parameters per instance.
| Component | Calculation | Parameters |
|---|---|---|
| Weight matrix | 2,359,296 | |
| Output bias | 768 | |
| Per-instance total | 2,360,064 | |
The reference module holds the weight output by input, [out, in].
Across 6 independent blocks: parameters.
Derived from the configured matrix dimensions; checkpoint tensors have not been independently counted.
Sources1
Direct evidence
The Perceiver blocks’ MLP: LayerNorm, an affine layer, exact GELU, dropout, an affine layer.
The implementation, pinned at revision e18c42acdfb7, not yet public