Hybridný, riedky model: 34 vrstiev, 128 expertov
Tri druhy vrstiev, jedna opakujúca sa jednotka
- 14 pamäť · Gated DeltaNet: stav lineárnej pozornosti na hlavu, aktualizovaný token po tokene
- 13 okno · pozornosť na posledných 512 tokenov, pozície RoPE
- 7 globálna · plná pozornosť, žiadne pozičné kódovanie; posledná vrstva je vždy globálna
Zmes expertov v každej vrstve
- 4 zo 128 smerovaných expertov vybraných na každý token smerovačom
- 1 spoločný expert je vždy aktívny
- 0,6 mld. zo 7,2 mld. parametrov aktívnych na token; veľkosť skrytej vrstvy 1 024, prepojené vstupné a výstupné vloženia
Optimalizátor
Muon + AdamW
Každá aktualizácia momentu sa pred jej aplikáciou ortogonalizuje (Newton–Schulz), takže každý smer matice váh sa pohybuje podobnou rýchlosťou, namiesto toho, aby dominovalo len niekoľko smerov.
Kontextové okno s 4 096 tokenmi počas trénovania; slovná zásoba s 151 643 tokenmi.