Chex

What's Changed
* Avoid use of deprecated `device_buffer` attriutes of jax.Array by copybara-service in https://github.com/google-deepmind/chex/pull/325
* Update test.sh by copybara-service in https://github.com/google-deepmind/chex/pull/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 https://github.com/google-deepmind/chex/pull/333
* Adjust test tolerances for test_assert_trees_all_close by copybara-service in https://github.com/google-deepmind/chex/pull/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 https://github.com/google-deepmind/chex/pull/336
* Add warnings in chex. by copybara-service in https://github.com/google-deepmind/chex/pull/337
* Add Dimensions.size to get flat size from named shape. by copybara-service in https://github.com/google-deepmind/chex/pull/339
* Increase version number to make a new release. by copybara-service in https://github.com/google-deepmind/chex/pull/342


**Full Changelog**: https://github.com/google-deepmind/chex/compare/v0.1.85...v0.1.86

What's Changed
* [JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu"). by copybara-service in https://github.com/google-deepmind/chex/pull/317


**Full Changelog**: https://github.com/google-deepmind/chex/compare/v0.1.84...v0.1.85

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

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

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`.