JAX v0.3.20

v0.3.20

JAX release v0.3.20

Added 1
  • Add missing .pyi files
Fixed 2
  • Fix incompatibility between jax 0.3.19 and the libtpu version it pinned
  • Fix incorrect pip url in setup.py comment

Notable changes:

  • Adds missing .pyi files that were missing from the previous release (#12536).
  • Fixes an incompatibility between jax 0.3.19 and the libtpu version it pinned (#12550). Requires jaxlib 0.3.20.
  • Fix incorrect pip url in setup.py comment (#12528).
View original

Upgraded? How did it go?

Discussion