Chex 0.1.2

hbq1 released this 31 Mar 17:04

e080c85

What's Changed

Support JAX parallel operations in chex.fake_pmap contexts by @copybara-service in #142
Remove references to jax.numpy.lax_numpy. by @copybara-service in #150

Full Changelog: v0.1.1...v0.1.2

Assets 2