Skip to content

Keep balanced routing auxiliary-loss gradients finite - #224

Merged
jpuigcerver merged 1 commit into
google-research:mainfrom
sylvesterkaczmarek:fix/finite-balanced-routing-gradients
Oct 15, 2026
Merged

jpuigcerver merged 1 commit into
google-research:mainfrom
sylvesterkaczmarek:fix/finite-balanced-routing-gradients

Commits

  1. Commits on Oct 3, 2026