Chex 0.1.91

What's Changed

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

Full Changelog: v0.1.90...v0.1.91

Chex 0.1.90

What's Changed

• Fix source code links to point to master branch. by @copybara-service[bot] in #384
• chex: avoid use of deprecated jax.util.unzip2 by @copybara-service[bot] in #391
• Update JAX nightly index usage by @copybara-service[bot] in #392
• [chex] remove stale try/except import by @copybara-service[bot] in #396
• Update restrict_backends after jax-ml/jax@0644886. by @copybara-service[bot] in #397
• Use backend_compile_and_load if available, and backend_compile otherwise. by @copybara-service[bot] in #398
• Support BF16 in assert_trees_all_close_ulp. by @copybara-service[bot] in #399
• Fixed typo. by @copybara-service[bot] in #400
• Release v0.1.90 by @copybara-service[bot] in #402

Full Changelog: v0.1.89...v0.1.90

Chex 0.1.89

What's Changed

• Switch to new unittest test loader API by @daskol in #372
• Add test against JAX nightly by @copybara-service in #376
• CI: test Python 3.12 by @copybara-service in #377
• Chex: add Python 3.13 CI tests by @copybara-service in #378
• Fix typo: "assetion" -> "assertion" by @copybara-service in #379
• Maintain compatibility with Python 3.13 by @daskol in #375

New Contributors

• @daskol made their first contribution in #372

Full Changelog: v0.1.88...v0.1.89

Chex 0.1.88

What's Changed

• Fix test_assert_trees_all_equal_prng_keys by @copybara-service in #366
• Avoid relying on jax.lib.xla_bridge._backends, as this is private and will soon be deleted by @copybara-service in #367
• Release 0.1.88. by @copybara-service in #369

Full Changelog: v0.1.87...v0.1.88

Chex 0.1.87

What's Changed

• Add flattening functionality to chex.Dimensions. by @copybara-service in #341
• Fix doc-string in warn_deprecated_function. by @copybara-service in #343
• Add utility to create a deprecated alias. by @copybara-service in #345
• Fix linter's warnings. by @copybara-service in #349
• Test RTD webhooks. by @copybara-service in #350
• Remove a bad async chexify test as the following is not guaranteed: by @copybara-service in #353
• Replace deprecated jax.tree_* functions with jax.tree.* by @copybara-service in #354
• Disable registration of dataclasses defined in main with JAX tree util. by @copybara-service in #356
• Chex: Add CI job for oldest supported JAX version by @copybara-service in #358
• make replacement kwarg in warn_deprecated_function optional by @copybara-service in #362
• increase version number for a pypi release by @copybara-service in #363

Full Changelog: v0.1.86...v0.1.87

Chex 0.1.86

What's Changed

• Avoid use of deprecated device_buffer attriutes of jax.Array by @copybara-service in #325
• Update test.sh by @copybara-service in #331
• Update chex.assert_type to check concrete types instead of just asserting that the type is a floating/integer sub-type. by @copybara-service in #333
• Adjust test tolerances for test_assert_trees_all_close by @copybara-service in #334
• Register dataclass at definition so that unpickling the dataclass object, type, and treedef works in most cases (with one exception). by @copybara-service in #336
• Add warnings in chex. by @copybara-service in #337
• Add Dimensions.size to get flat size from named shape. by @copybara-service in #339
• Increase version number to make a new release. by @copybara-service in #342

Full Changelog: v0.1.85...v0.1.86

Chex 0.1.85

What's Changed

• [JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu"). by @copybara-service in #317

Full Changelog: v0.1.84...v0.1.85

Chex 0.1.84

Update the type definitions in pytype.py and fix various warnings.

Chex 0.1.83

• Remove deprecated assert_tree_* methods;
• Improve the usability of dataclasses;
• Update dependencies and pytypes in accordance with the latest changes in Jax.

Chex 0.1.82

What's Changed

Chex has moved from external dm-tree to the JAX-native jax.tree_util library for operating with PyTrees.
This means that Nones will not be treated as separate leaves by chex tree assertions any longer.
Note that you can still check for Nones using chex.assert_tree_no_nones().

• Remove usage of dm-tree in asserts. by @copybara-service in #288
• Add support for kw_only for chex.dataclass. by @jblespiau in #290
• Deprecate ignore_nones. by @copybara-service in #292

Full Changelog: v0.1.81...v0.1.82