Latent encoder
Output RMSNorm
×learned
Rescale each position’s 768 features by their root mean square and apply a learned gain.
- in
- [B batch, T positions, 768 d]
- out
- [B batch, T positions, 768 d]
- parameters
- 768
ε bounds the divisor from below; it is not added under the root.
Parameters
768 parameters owned by this component
| Component | Calculation | Parameters |
|---|---|---|
| RMSNorm gain | 768 | |
| Per-instance total | 768 | |
Derived from the pinned implementation’s constructors and configuration. The checkpoint’s tensors have not been counted. 306,011,162 of these parameters are trained; frozen GPT-2 and the decoder’s mask vector, 124,442,112, are not.
One gain for each of the 768 features.
Sources3
Direct evidence
RMSNorm divides by the vector’s norm times width^(−1/2), clamped from below at ε, and multiplies by a learned gain.
The implementation, pinned at revision e18c42acdfb7, not yet public
Direct evidence
The encoder owns the learned latent queries, a position table, an input and an output RMSNorm and the blocks.
The implementation, pinned at revision e18c42acdfb7, not yet public
Direct evidence
The encoder adds positions to the features, normalizes the learned queries, applies the blocks, each attending to the features, and normalizes the result.
The implementation, pinned at revision e18c42acdfb7, not yet public