Chex 0.1.8
What's Changed
- Drop support for python 3.8, as per JAX deprecation policy.
- Use new
jax.tree_util.register_pytree_with_keys
to registerchex.Dataclass
. by @copybara-service in #266 - Add size based assertions by @tttc3 in #265
- Integer sequence support for
chex.assert_tree_shape{_prefix, _suffix}
by @hylkedonker in #275 - Resolve unsoundness caught by pytype --strict-none-binding. by @copybara-service in #276
- Release v0.1.8. by @copybara-service in #281
New Contributors
- @tttc3 made their first contribution in #265
- @hylkedonker made their first contribution in #275
Full Changelog: v0.1.7...v0.1.8