Ulysses Sequence Parallelism: Training with Million-Token Contexts

Ulysses Sequence Parallelism (SP) distribue le calcul de l'attention sur plusieurs GPU en fendant la séquence et les heads d'attention, permettant un entraînement avec des contextes de plusieurs millions de tokens. Chaque GPU traite un shard de séquence puis procède à des all-to-all pour échanger les clés/valeurs par sous-ensembles de heads. Cette approche du Snowflake AI Research intègre nativement Hugging Face (Accelerate, Transformers Trainer, TRL SFTTrainer).

Source

HuggingFace Blog

Ingéré

09 MARS 2026 · 19:10

Score édito

5.0 / 5