Hybridní, řídce parametrizovaný model: 34 vrstev, 128 expertů
Tři druhy vrstev, jedna opakující se jednotka
- 14 paměť · Gated DeltaNet: stav lineární pozornosti na hlavu, aktualizovaný token po tokenu
- 13 okno · pozornost zaměřená na posledních 512 tokenů, pozice RoPE
- 7 globální · plná pozornost, žádné poziční kódování; poslední vrstva je vždy globální
Mixture of experts v každé vrstvě
- 4 ze 128 směrovaných expertů vybraných pro každý token směrovačem
- 1 sdílený expert je vždy aktivní
- 0,6 mld. ze 7,2 mld. parametrů aktivních na token; skrytá velikost 1 024, propojená vstupní a výstupní vložení
Optimalizátor
Muon + AdamW
Každá aktualizace momentu je před aplikací ortogonalizována (Newton–Schulz), takže se každý směr matice vah pohybuje podobnou rychlostí, namísto toho, aby dominovalo jen několik směrů.
Kontextové okno o velikosti 4 096 tokenů během tréninku; slovní zásoba o velikosti 151 643 tokenů.