Skip to content

JAX v0.5.2

Choose a tag to compare

View all tags
@skye skye released this 05 Mar 02:36
· 6208 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