JAX release v0.3.20

Notable changes:

• Adds missing .pyi files that were missing from the previous release (#12536).
• Fixes an incompatibility between jax 0.3.19 and the libtpu version it pinned (#12550). Requires jaxlib 0.3.20.
• Fix incorrect pip url in setup.py comment (#12528).

JAX release v0.3.19

Fixes the required jaxlib version

JAX release v0.3.18

• GitHub commits.
• Changes
  • Ahead-of-time lowering and compilation functionality (tracked in
    {jax-issue}#7733) is stable and public. See the
    overview and the API docs
    for {mod}jax.stages.
  • Introduced {class}jax.Array, intended to be used for both isinstance checks
    and type annotations for array types in JAX. Notice that this included some subtle
    changes to how isinstance works for {class}jax.numpy.ndarray for jax-internal
    objects, as {class}jax.numpy.ndarray is now a simple alias of {class}jax.Array.
• Breaking changes
  • jax._src is no longer imported into the from the public jax namespace.
    This may break users that were using JAX internals.
  • jax.soft_pmap has been deleted. Please use pjit or xmap instead.
    jax.soft_pmap is undocumented. If it were documented, a deprecation period
    would have been provided.

JAX release v0.3.17

• GitHub commits.
• Bugs
  • Fix corner case issue in gradient of lax.pow with an exponent of zero (#12041)
• Breaking changes
  • jax.checkpoint, also known as jax.remat, no longer supports the concrete option, following the previous version's deprecation; see JEP 11830.
• Changes
  • Added jax.pure_callback that enables calling back to pure Python functions from compiled functions (e.g. functions decorated with jax.jit or jax.pmap).
• Deprecations:
  • The deprecated DeviceArray.tile() method has been removed. Use jax.numpy.tile (#11944).
  • DeviceArray.to_py() has been deprecated. Use np.asarray(x) instead.

JAX release v0.3.16

• GitHub commits.
• Breaking changes
  • Support for NumPy 1.19 has been dropped, per the
    deprecation policy.
    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
• 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.

jaxlib version 0.3.15

jaxlib-v0.3.15

jaxlib version 0.3.15

jax version 0.3.15

jax-v0.3.15

jax version 0.3.15

Jaxlib release v0.3.14

jaxlib-v0.3.14

Jaxlib release v0.3.14

JAX release v0.3.14

jax-v0.3.14

JAX release v0.3.14

JAX release v0.3.13

jax-v0.3.13

jax version 0.3.13