# JAX v0.3.16 — JAX release v0.3.16

- Product: JAX (https://whatsnew.fyi/product/jax)
- Vendor: Google
- Date: 2022-08-12
- Version: v0.3.16
- Original notes: https://github.com/jax-ml/jax/releases/tag/jax-v0.3.16
- Permalink: https://whatsnew.fyi/product/jax/releases/v0.3.16

What's New is an index, not a publisher: every entry below links to the vendor's own release notes, which are the authoritative source. Entries are labelled where they are hand-curated sample data, pre-releases, or drawn from a secondary source such as a developer blog.

Reuse: the summaries, labels and curation here are © What's New. Quote freely with attribution and a link back; wholesale republication of the corpus is not permitted — terms: https://whatsnew.fyi/terms. The vendors' own release notes remain their publishers'.

---

- **removed** — Support for NumPy 1.19 has been dropped
- **added** — Add jax.debug module with utilities for runtime value debugging including jax.debug.print and jax.debug.breakpoint
- **removed** — Remove jax.mask and jax.shapecheck APIs
- **removed** — Remove jax.experimental.loops
- **removed** — Remove jax.tree_util.tree_multimap
- **removed** — Remove jax.experimental.stax deprecated alias
- **removed** — Remove jax.experimental.optimizers deprecated alias
- **changed** — Replace jax.checkpoint and jax.remat default implementation with new implementation, deprecating the old one

* [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.15...main).
* Breaking changes
  * Support for NumPy 1.19 has been dropped, per the
    [deprecation policy](https://jax.readthedocs.io/en/latest/deprecation.html).
    Please upgrade to NumPy 1.20 or newer.
* Changes
  * Added `jax.debug` that includes utilities for runtime value debugging such at `jax.debug.print` and `jax.debug.breakpoint`.
  * Added new documentation for [runtime value debugging](https://jax.readthedocs.io/en/latest/debugging/index.html)
* 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, and `jax.tree_util.tree_map` is a direct replacement.
  * Removed `jax.experimental.stax`; it has long been a deprecated alias of `jax.example_libraries.stax`.
  * Removed `jax.experimental.optimizers`; it has long been a deprecated alias of `jax.example_libraries.optimizers`.
  * `jax.checkpoint`, also known as `jax.remat`, has a new implementation switched on by default, meaning the old implementation is deprecated; see [JEP 11830](https://jax.readthedocs.io/en/latest/jep/11830-new-remat-checkpoint.html).