The Wire · Showcase
Transformers MoE loss cut 99.7% as TRL gains tool calling
By RepoJournal · Filed · About Hugging Face
A single PR eliminates a 128k-token memory bomb in MoE training while TRL's distillation trainer learns to call tools.
Transformers slashed MoE load-balancing loss memory by 99.7% at 128k tokens by computing per-layer instead of materializing a giant one-hot tensor [1]. That fix lands alongside a nemotron_h save_pretrained correction that restores the plural embedding key for all published checkpoints [2]. TRL's DistillationTrainer now supports tool calling, mirroring GRPO with masked tool-result tokens and a configurable iteration cap [3]. The same author reorganized TRL examples into per-task folders, untangling scripts from notebooks [4]. Kernels-community ported SageAttention3 Blackwell FP4 attention, tested on RTX-PRO-6000, bringing microscaling NVFP4 to sm_120 [5]. But watch the kernels telemetry doc fix: a typo in the title hides a real improvement in env var documentation [6]. The SFT collator now emits position_ids for sequence parallelism, fixing a crash on the first batch in ALST/Ulysses setups [7]. That's the one to test before you trust it at scale.
One email a day. Unsubscribe in one click.
Keep up with Hugging Face in about 3 minutes a day: what actually shipped — the commits, pull requests, releases, and security advisories that matter.
One email a day. Unsubscribe in one click. Read a past issue →
Action items
- → Cherry-pick the MoE load-balancing loss fix before your next long-context training run huggingface/transformers [immediate]
- → Review and update nemotron_h conversion scripts to expect plural embedding keys huggingface/transformers [plan]
- → Test DistillationTrainer tool-calling on your own workflow huggingface/trl [monitor]
- → Run SFT with sequence parallelism after the collator fix huggingface/trl [monitor]
References
- [1] Compute MoE load-balancing loss per layer to avoid giant one-hot materialization −99.7% @ 128k ↗ huggingface/transformers
- [2] Fix nemotron_h save_pretrained emitting singular backbone.embedding.weight (#48075) huggingface/transformers
- [3] [DistillationTrainer refactor] Support tool calling ↗ huggingface/trl
- [4] Reorganize examples into self-contained per-example folders ↗ huggingface/trl
- [5] sage-blackwell: port SageAttention3 Blackwell FP4 attention ↗ huggingface/kernels-community
- [6] chore: document telemetry env var beter. ↗ huggingface/kernels
- [7] Emit position_ids from the SFT collator when sequence parallelism is enabled ↗ huggingface/trl