-
New features:
- Added
jax.tree.broadcastwhich implements a pytree prefix broadcasting helper.
- Added
-
Changes
- The minimum NumPy version is 1.26 and the minimum SciPy version is 1.12.
New features:
jax.tree.broadcast which implements a pytree prefix broadcasting helper.Changes