Chex 0.1.85

hbq1 released this 22 Nov 18:26

· 27 commits to master since this release

What's Changed

[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu"). by @copybara-service in #317

Full Changelog: v0.1.84...v0.1.85

Assets 2