Chex 0.1.1

What's Changed

• Move dataclass registration to init so that it's invoked after deserialization. by @copybara-service in #111
• Add pytype for jax array's dtypes. by @copybara-service in #112
• Fix dataclass registration on deserialization. by @copybara-service in #114
• Fix restrict_backends after jax.xla.backend_compile was moved by @copybara-service in #116
• Refactor asserts.py and warn users not to rely on asserts_internal's functionality. by @copybara-service in #117
• Set up ReadTheDoc pages and add a few examples. by @copybara-service in #118
• Include Sphinx builds into CI tests. by @copybara-service in #119
• Adds internal functionality by @copybara-service in #122
• Update Chex citation. by @copybara-service in #125
• Refactor assertions in preparation for including them into the RTD docs. by @copybara-service in #126
• Add asserts, variants, and pytypes modules to the RTD docs. by @copybara-service in #127
• Fix references to collections.abc.Mappable -> collections.abc.Mapping in docs and comments. collections.abc.Mappable does not exist. by @copybara-service in #129
• Document the rational behing the mappability of chex.dataclasses. by @copybara-service in #130
• Add 3 new tree assertions: by @copybara-service in #131
• Add assert_tree_is_sharded for asserting that a tree is sharded across the specified devices. by @copybara-service in #132
• Add PyTreeDef to pytypes. by @copybara-service in #134
• Disallow ShardedDeviceArrays in assert_tree_is_on_host and assert_tree_is_on_device. by @copybara-service in #133
• Bump ipython from 7.16.1 to 7.16.3 in /requirements by @dependabot in #135
• Remove the old venv directory before testing the package. by @copybara-service in #138
• Refactor asserts.py in preparation for experimental device assertions. by @copybara-service in #137
• Fix minor typo in docs. by @copybara-service in #139
• Improve exception message for assert_tree_shape_prefix. by @copybara-service in #143
• Release v0.1.1 by @copybara-service in #146

Full Changelog: v0.1.0...v0.1.1