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

Keep balanced routing auxiliary-loss gradients finite

11e37d5
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Oct 3, 2026 in 1s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

✅ 11e37d5 Author: @sylvesterkaczmarek <16242628+sylvesterkaczmarek​@users.noreply.github.com>

(Only the first commit for a unique contributor is listed.)