Skip to content

Avoid NaN gradients for balanced routing losses - #218

Merged
jpuigcerver merged 1 commit into
google-research:mainfrom
sylvesterkaczmarek:fix-balanced-routing-gradients
Sep 11, 2026
Merged

Avoid NaN gradients for balanced routing losses#218
jpuigcerver merged 1 commit into
google-research:mainfrom
sylvesterkaczmarek:fix-balanced-routing-gradients

Avoid NaN gradients for balanced routing losses

fcf0d35
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Sep 10, 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:

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

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

View more details on Google CLA