Un modèle hybride et épars : 34 couches, 128 experts
Trois types de couches, une unité répétitive
- 14 mémoire · Gated DeltaNet : un état d’attention linéaire par tête, mis à jour token par token
- 13 fenêtre · attention portée aux 512 derniers tokens, positions RoPE
- 7 global · attention complète, pas de codage positionnel ; la dernière couche est toujours globale
Mixture of experts dans chaque couche
- 4 experts sur 128, acheminés et choisis par un routeur pour chaque token
- 1 expert partagé toujours actif
- 0,6 milliard de paramètres actifs sur 7,2 milliards par token ; taille des couches cachées : 1 024 ; embeddings d’entrée et de sortie liés
Optimiseur
Muon + AdamW
Chaque mise à jour du momentum est orthogonalisée (Newton–Schulz) avant d’être appliquée, de sorte que toutes les directions d’une matrice de poids évoluent à un rythme similaire, au lieu que quelques-unes dominent.
Fenêtre de contexte de 4 096 tokens lors de l’entraînement ; vocabulaire de 151 643 tokens.