JAX v0.6.2

v0.6.2

JAX v0.6.2

Added 1
  • Added jax.tree.broadcast which implements a pytree prefix broadcasting helper
Changed 1
  • The minimum NumPy version is 1.26 and the minimum SciPy version is 1.12

  • New features:

    • Added jax.tree.broadcast which implements a pytree prefix broadcasting helper.

  • Changes

    • The minimum NumPy version is 1.26 and the minimum SciPy version is 1.12.
View original

Upgraded? How did it go?

Discussion