Skip to content

Chex 0.1.6
Compare
Choose a tag to compare
@hbq1 hbq1 released this 06 Feb 12:09
· 102 commits to master since this release
v0.1.6
2008375

What's Changed

  • Replace jnp.DeviceArray with the new public type jax.Array. by @copybara-service in #187
  • Unpack assertions' kwargs at a higher level for improved readability. by @copybara-service in #209
  • Allow calling @chexify-ed functions from non-main threads. by @copybara-service in #214
  • Add assert_trees_all_equal_shapes_and_dtypes by @copybara-service in #219
  • Implement a new unflatten for chex.dataclass that avoids init while keeping the (un)flattened order unchanged. by @copybara-service in #222
  • Fix examples in the docs & linter warnings. by @copybara-service in #235
  • Replace uses of jax.lib.xla_extension.Device with jax.Device. by @copybara-service in #237

Full Changelog: v0.1.5...v0.1.6

Assets 2
Loading