Skip to content

JAX v0.5.2

Latest
Latest
Compare
Choose a tag to compare
@skye skye released this 05 Mar 02:36
· 290 commits to main since this release
jax-v0.5.2
ce22429

Patch release of 0.5.1

  • Bug fixes
    • Fixes TPU metric logging and tpu-info, which was broken in 0.5.1
Assets 2
Loading