- GitHub commits.
- Breaking changes
- Support for NumPy 1.19 has been dropped, per the
deprecation policy.
Please upgrade to NumPy 1.20 or newer.
- Support for NumPy 1.19 has been dropped, per the
- Changes
- Added
jax.debug
that includes utilities for runtime value debugging such atjax.debug.print
andjax.debug.breakpoint
. - Added new documentation for runtime value debugging
- Added
- Deprecations
jax.mask
jax.shapecheck
APIs have been removed. See #11557.jax.experimental.loops
has been removed. See #10278 for an alternative API.jax.tree_util.tree_multimap
has been removed. It has been deprecated since JAX release 0.3.5, andjax.tree_util.tree_map
is a direct replacement.- Removed
jax.experimental.stax
; it has long been a deprecated alias ofjax.example_libraries.stax
. - Removed
jax.experimental.optimizers
; it has long been a deprecated alias ofjax.example_libraries.optimizers
. jax.checkpoint
, also known asjax.remat
, has a new implementation switched on by default, meaning the old implementation is deprecated; see JEP 11830.