google-deepmind/distrax v0.1.6

Distrax 0.1.6

Repository: google-deepmind/distrax

Tag: v0.1.6

Published: 2025-09-01T20:33:04Z

Prerelease: no

Release notes:

What's Changed

• The main tensorflow_probability BUILD target is being changed to only load the by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/272
• Add sqrt/square primitives to the inverse registry after https://github.com/openxla/stablehlo/pull/2623 by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/279
• Remove references to jax.core.raise_to_shaped by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/281
• Add accuracy field to unary ops by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/285
• Configure tests to explicitly use jax_threefry_partitionable=False. by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/286
• Configure tests to explicitly use jax_threefry_partitionable=False. by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/288
• [distrax] Replace usages of deprecated jax.interpreters.xla.abstractify by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/289
• remove pjit_p, leaving only jax.extend.core.primitives.jit_p by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/293
• Disable bad-ident warnings. by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/300
• release v0.1.6 by @copybara-service[bot] in https://github.com/google-deepmind/distrax/pull/302

Full Changelog: https://github.com/google-deepmind/distrax/compare/v0.1.42...v0.1.6