google-deepmind/distrax v0.1.8

v0.1.8

Repository: google-deepmind/distrax

Tag: v0.1.8

Published: 2026-04-16T21:25:16Z

Prerelease: no

Release notes:

What's Changed

• Migrate from jax.experimental.enable_x64 to jax.enable_x64. by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/307
• Replace deprecated jax.core.get_aval with jax.typeof by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/319
• [JAX] Remove stale references to //jax:experimental, which is being removed. by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/322
• Fix Distrax from Jax API change. by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/325

Full Changelog: https://github.com/google-deepmind/distrax/compare/v0.1.7...v0.1.8