Chex

What's Changed
* Fix `test_assert_trees_all_equal_prng_keys` by copybara-service in https://github.com/google-deepmind/chex/pull/366
* Avoid relying on jax.lib.xla_bridge._backends, as this is private and will soon be deleted by copybara-service in https://github.com/google-deepmind/chex/pull/367
* Release 0.1.88. by copybara-service in https://github.com/google-deepmind/chex/pull/369


**Full Changelog**: https://github.com/google-deepmind/chex/compare/v0.1.87...v0.1.88

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


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

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.