Un modello ibrido e sparso: 34 strati, 128 esperti
Tre tipi di layer, un'unità ripetuta
- 14 memoria · Gated DeltaNet: uno stato di attenzione lineare per head, aggiornato token per token
- 13 finestra · attenzione sugli ultimi 512 token, posizioni RoPE
- 7 globale · attenzione completa, nessuna codifica posizionale; l'ultimo livello è sempre globale
Mixture of experts in ogni livello
- 4 di 128 esperti instradati scelti per token da un router
- 1 esperto condiviso sempre attivo
- 0,6B di 7,2B parametri attivi per token; dimensione nascosta 1.024, embedding di input e output vincolati
Ottimizzatore
Muon + AdamW
Ogni aggiornamento del momentum viene ortogonalizzato (Newton–Schulz) prima di essere applicato, in modo che tutte le direzioni di una matrice dei pesi si muovano a una velocità simile, anziché lasciarne alcune a dominare.
Finestra di contesto di 4.096 token durante l'addestramento; vocabolario di 151.643 token.