google-deepmind/chex v0.1.91

Chex 0.1.91

Repository: google-deepmind/chex

Tag: v0.1.91

Published: 2025-09-01T21:48:43Z

Prerelease: no

Release notes:

What's Changed

• Migrate references to jax.lib.xla_extension to jax.extend. by @copybara-service[bot] in https://github.com/google-deepmind/chex/pull/403
• Add assert_tree_shape, which is like assert_shape, but for all leaves in a tree. by @copybara-service[bot] in https://github.com/google-deepmind/chex/pull/404
• Use format_shape_matcher when constructing the assert_tree_shape error message. by @copybara-service[bot] in https://github.com/google-deepmind/chex/pull/405
• Migrate to pyproject.toml by @copybara-service[bot] in https://github.com/google-deepmind/chex/pull/406
• release v0.1.91 by @copybara-service[bot] in https://github.com/google-deepmind/chex/pull/407

Full Changelog: https://github.com/google-deepmind/chex/compare/v0.1.90...v0.1.91