Latent encoder
Learned latent queries
×learned
1,024 learned vectors, the same for every sequence, repeated over the batch.
- out
- [B batch, T positions, 768 d]
- parameters
- 786,432
Parameters
786,432 parameters owned by this component
| Component | Calculation | Parameters |
|---|---|---|
| table | 786,432 | |
| Per-instance total | 786,432 | |
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.
A table of 1,024 rows of 768 features, initialized with standard deviation 0.02.
Explanation
The table is a parameter read whole, not looked up by token: row becomes the latent of position .
Sources2
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