Chex 0.1.4

What's Changed

• Add an InitVar field in the dataclass tests. by @copybara-service in #161
• Download latest .pylintrc version in tests. by @copybara-service in #167
• Fix assert_axis_dimension_comparator usages. by @copybara-service in #168
• Update "jax.tree_util" functions by @copybara-service in #171
• Use jax.tree_util.tree_map in place of deprecated tree_multimap. by @copybara-service in #175
• Silence some pytype errors. by @copybara-service in #174
• Add chex.Dimensions utility for readable shape asserts. by @copybara-service in #169

Full Changelog: v0.1.3...v0.1.4