Chex 0.1.92

What's Changed

• update docs builder by @copybara-service[bot] in #410
• [pmap] Update logic that works with both PmapSharding and NamedSharding. by @copybara-service[bot] in #414
• [pmap] Avoid degraded performance under the new jax.pmap. by @copybara-service[bot] in #419
• Fix a deprecation warning for new versions of JAX and Numpy. Fixes #424. by @copybara-service[bot] in #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 #436
• [pmap] Remove jax.config.pmap_shmap_merge. by @copybara-service[bot] in #438
• [pmap] In-line definitions of jax.device_put_sharded and jax.device_put_replicated. by @copybara-service[bot] in #440
• fix tests and drop python 3.11 by @copybara-service[bot] in #442
• Release v0.1.92 by @copybara-service[bot] in #443

Full Changelog: v0.1.91...v0.1.92