JAX v0.4.32

v0.4.32

JAX release v0.4.32

Security 1
  • Fixed data corruption bug on TPU when multiple TPU slices are present in the job

WARNING: This release has been yanked from PyPI because of a data corruption bug on TPU if there are multiple TPU slices in the job

View original

Upgraded? How did it go?

Discussion