Skip to content

Chex 0.0.9

Choose a tag to compare

View all tags
@hbq1 hbq1 released this 16 Nov 19:08
v0.0.9
1245d00

It is the latest version compatible with Python 3.6. See google-deepmind/optax#222 for more details.

Changes since 0.0.8:

  • Use rtol=1e-6 in asserts.assert_tree_close;
  • Added asserts.assert_trees_all_equal;
  • Removed restricted_inheritance option from Chex dataclasses;
  • Added dims= option to assert_equal_shape, to check a subset of dims;
  • Added test.sh for launching CI tests on a local machine;
  • Added support for default exception messages and types to assertions;
  • Added support for jnp.bfloat16 to asserts.assert_trees_all_close();
  • Added support for static_argnames to variants.with_jit;
  • Added a restrict_backends module for constraining the set of backends that a region of code can use;
  • Added asserts.assert_trees_all_equal_dtypes assertion;
  • Exposed asserts.assert_tree_shape_suffix to the public API;
  • Added asserts.assert_tree_shape_suffix to check whether arrays share the same suffix.
Assets 2
Loading