Chex

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

* Remove usage of dm-tree in asserts. by copybara-service in https://github.com/deepmind/chex/pull/288
* Add support for `kw_only` for chex.dataclass. by jblespiau in https://github.com/deepmind/chex/pull/290
* Deprecate `ignore_nones`. by copybara-service in https://github.com/deepmind/chex/pull/292


**Full Changelog**: https://github.com/deepmind/chex/compare/v0.1.81...v0.1.82

Update requirements.txt to use `numpy>=1.25.0`.

**Full Changelog**: https://github.com/deepmind/chex/compare/v0.1.4...v0.1.5

What's Changed
* Drop support for python 3.8, as per JAX deprecation policy.
* Use new `jax.tree_util.register_pytree_with_keys` to register `chex.Dataclass`. by copybara-service in https://github.com/deepmind/chex/pull/266
* Add size based assertions by tttc3 in https://github.com/deepmind/chex/pull/265
* Integer sequence support for `chex.assert_tree_shape{_prefix, _suffix}` by hylkedonker in https://github.com/deepmind/chex/pull/275
* Resolve unsoundness caught by pytype --strict-none-binding. by copybara-service in https://github.com/deepmind/chex/pull/276
* Release v0.1.8. by copybara-service in https://github.com/deepmind/chex/pull/281

New Contributors
* tttc3 made their first contribution in https://github.com/deepmind/chex/pull/265
* hylkedonker made their first contribution in https://github.com/deepmind/chex/pull/275

**Full Changelog**: https://github.com/deepmind/chex/compare/v0.1.7...v0.1.8

New Contributors
* SauravMaheshkar made their first contribution in https://github.com/deepmind/chex/pull/239
* pfackeldey made their first contribution in https://github.com/deepmind/chex/pull/241

**Full Changelog**: https://github.com/deepmind/chex/compare/v0.1.6...v0.1.7

What's Changed
* Replace `jnp.DeviceArray` with the new public type `jax.Array`. by copybara-service in https://github.com/deepmind/chex/pull/187
* Unpack assertions' kwargs at a higher level for improved readability. by copybara-service in https://github.com/deepmind/chex/pull/209
* Allow calling `chexify`-ed functions from non-main threads. by copybara-service in https://github.com/deepmind/chex/pull/214
* Add `assert_trees_all_equal_shapes_and_dtypes` by copybara-service in https://github.com/deepmind/chex/pull/219
* Implement a new unflatten for chex.dataclass that avoids __init__ while keeping the (un)flattened order unchanged. by copybara-service in https://github.com/deepmind/chex/pull/222
* Fix examples in the docs & linter warnings. by copybara-service in https://github.com/deepmind/chex/pull/235
* Replace uses of jax.lib.xla_extension.Device with jax.Device. by copybara-service in https://github.com/deepmind/chex/pull/237


**Full Changelog**: https://github.com/deepmind/chex/compare/v0.1.5...v0.1.6