Skip to content

fix(distrax): Tanh NaN clip + MultivariateNormal vmap batch_shape fix - #328

Open
Sumu004 wants to merge 2 commits into
google-deepmind:mainfrom
Sumu004:fix/tanh-bijector-nan-clip
Open

fix(distrax): Tanh NaN clip + MultivariateNormal vmap batch_shape fix#328
Sumu004 wants to merge 2 commits into
google-deepmind:mainfrom
Sumu004:fix/tanh-bijector-nan-clip

fix(mvn_from_bijector): make batch_shape robust to vmap-prepended dim…

e222311
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Jul 11, 2026 in 7s

✅ 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:

✅ e222311 Author: @Sumu004 <sumu********araj​@gmail.com>

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

View more details on Google CLA