jax-v0.6.2
Breaking Changes📦 jaxView on GitHub →
⚠ 1 breaking✨ 1 features🔧 1 symbols
Summary
This release introduces the new `jax.tree.broadcast` helper and raises the minimum required versions of NumPy and SciPy.
⚠️ Breaking Changes
- The library now requires NumPy >=1.26 and SciPy >=1.12; environments using older versions must upgrade those dependencies.
✨ New Features
- Added `jax.tree.broadcast`, a pytree prefix broadcasting helper.
🔧 Affected Symbols
jax.tree.broadcast