Chex

What's Changed
* Add support for value assertions in jitted functions. by copybara-service in https://github.com/deepmind/chex/pull/178
* [JAX] Avoid private implementation detail _ScalarMeta. by copybara-service in https://github.com/deepmind/chex/pull/180
* [JAX] Avoid implicit references to jax._src. by copybara-service in https://github.com/deepmind/chex/pull/181

What's Changed
* Add an InitVar field in the dataclass tests. by copybara-service in https://github.com/deepmind/chex/pull/161
* Download latest .pylintrc version in tests. by copybara-service in https://github.com/deepmind/chex/pull/167
* Fix `assert_axis_dimension_comparator` usages. by copybara-service in https://github.com/deepmind/chex/pull/168
* Update "jax.tree_util" functions by copybara-service in https://github.com/deepmind/chex/pull/171
* Use jax.tree_util.tree_map in place of deprecated tree_multimap. by copybara-service in https://github.com/deepmind/chex/pull/175
* Silence some pytype errors. by copybara-service in https://github.com/deepmind/chex/pull/174
* Add chex.Dimensions utility for readable shape asserts. by copybara-service in https://github.com/deepmind/chex/pull/169


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

What's Changed
* Slight helping clarification to `clear_trace_counter`. by lucasb-eyer in https://github.com/deepmind/chex/pull/148
* Add new JAX-specific pytypes to chex pytypes. by copybara-service in https://github.com/deepmind/chex/pull/153
* Remove `chex.{C,G,T}puDevice` in favour of `chex.Device`. by copybara-service in https://github.com/deepmind/chex/pull/154

New Contributors
* lucasb-eyer made their first contribution in https://github.com/deepmind/chex/pull/148

**Full Changelog**: https://github.com/deepmind/chex/compare/v0.1.2...v0.1.3

What's Changed
* Support JAX parallel operations in chex.fake_pmap contexts by copybara-service in https://github.com/deepmind/chex/pull/142
* Remove references to jax.numpy.lax_numpy. by copybara-service in https://github.com/deepmind/chex/pull/150


**Full Changelog**: https://github.com/deepmind/chex/compare/v0.1.1...v0.1.2

**Full Changelog**: https://github.com/deepmind/chex/compare/v0.1.0...v0.1.1

Support for Python 3.6 has been dropped as per [JAX deprecation policy](https://jax.readthedocs.io/en/latest/deprecation.html). Please upgrade to a supported Python version.