Un model híbrid i dispers: 34 capes, 128 experts
Tres tipus de capes, una unitat repetitiva
- 14 memòria · Gated DeltaNet: un estat d'atenció lineal per cap, actualitzat tòquen per tòquen
- 13 finestra · atenció sobre els últims 512 tokens, posicions RoPE
- 7 global · atenció completa, sense codificació posicional; la darrera capa sempre és global
Barreja d'experts a cada capa
- 4 de 128 experts encaminats seleccionats per token per un encaminador
- 1 expert compartit sempre actiu
- 0,6 B de 7,2 B de paràmetres actius per testimoni; mida oculta de 1.024, incrustacions d'entrada i de sortida vinculades
Optimitzador
Muon + AdamW
Cada actualització de l'impuls s'ortogonalitza (Newton–Schulz) abans d'aplicar-se, de manera que totes les direccions d'una matriu de pesos es mouen a una velocitat similar, en lloc que n'hi hagi unes poques que dominin.
Finestra de context de 4.096 tokens durant l'entrenament; vocabulari de 151.643 tokens.