google-deepmind/distrax v0.1.9

Distrax 0.1.9

Repository: google-deepmind/distrax

Tag: v0.1.9

Published: 2026-06-12T23:38:54Z

Prerelease: no

Release notes:

What's Changed

• Avoid usage of deprecated jax.core APIs. by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/326
• [JAX] Replace uses of jnp.float_, jnp.int_, and jnp.complex_ with float, int, and complex in .astype, jnp.zeros, and jnp.ones. by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/327
• fix tests and drop python 3.11 by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/332
• Release v0.1.9 by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/333

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