google-deepmind/chex v0.1.92

Chex 0.1.92

Repository: google-deepmind/chex

Tag: v0.1.92

Published: 2026-06-12T14:27:45Z

Prerelease: no

Release notes:

What's Changed

• update docs builder by @copybara-service[bot] in https://github.com/google-deepmind/chex/pull/410
• [pmap] Update logic that works with both PmapSharding and NamedSharding. by @copybara-service[bot] in https://github.com/google-deepmind/chex/pull/414
• [pmap] Avoid degraded performance under the new jax.pmap. by @copybara-service[bot] in https://github.com/google-deepmind/chex/pull/419
• Fix a deprecation warning for new versions of JAX and Numpy. Fixes #424. by @copybara-service[bot] in https://github.com/google-deepmind/chex/pull/425
• Add strict mode support in chex.assert_trees_all_close. This aligns the behaviour with chex.assert_trees_all_equal. by @copybara-service[bot] in https://github.com/google-deepmind/chex/pull/436
• [pmap] Remove jax.config.pmap_shmap_merge. by @copybara-service[bot] in https://github.com/google-deepmind/chex/pull/438
• [pmap] In-line definitions of jax.device_put_sharded and jax.device_put_replicated. by @copybara-service[bot] in https://github.com/google-deepmind/chex/pull/440
• fix tests and drop python 3.11 by @copybara-service[bot] in https://github.com/google-deepmind/chex/pull/442
• Release v0.1.92 by @copybara-service[bot] in https://github.com/google-deepmind/chex/pull/443

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