Jax: jax-v0.10.1 Release

Release date:
May 18, 2026
Previous version:
jax-v0.10.0 (released April 16, 2026)
Magnitude:
31,116 Diff Delta
Contributors:
53 total committers
Data confidence:
Commits:

489 Commits in this Release

Ordered by the degree to which they evolved the repo in this version.

Authored April 23, 2026
Authored April 29, 2026

Top Contributors in jax-v0.10.1

hawkinsp
a-googler
yashk2810
superbobry
bchetioui
apaszke
golechwierowicz
allanrenucci
jakevdp
IvyZX

Directory Browser for jax-v0.10.1

All files are compared to previous version, jax-v0.10.0. Click here to browse diffs between other versions.

Loading File Browser...

Release Notes Published

  • New features

    • Added ResizeMethod.AREA to jax.image.resize, which matches TensorFlow's AREA resizing (#20098).
    • Added jax.scipy.linalg.hadamard for constructing Hadamard matrices (#10144).
    • Added jax.scipy.linalg.circulant for constructing circulant matrices (#10144).
    • Added jax.scipy.linalg.dft for constructing discrete Fourier transform matrices (#10144).
    • Added jax.scipy.linalg.leslie for constructing Leslie matrices (#10144).
    • Added jax.scipy.linalg.companion for constructing companion matrices from polynomial coefficients (#10144).
    • Added jax.scipy.linalg.fiedler for constructing symmetric Fiedler matrices (#10144).
    • Added jax.scipy.linalg.helmert for constructing Helmert matrices (#10144).
    • Moved RNG APIs from "implementations" to dtypes (#27854):
    • Added jax.random.key_dtype to get the dtype corresponding to a PRNG implementation name.
    • jax.random.key and wrap_key_data now accept a dtype argument.
  • Breaking changes

    • with mesh: context manager has been deprecated. Please use with jax.set_mesh(mesh): instead.
  • Deprecations

    • Passing the copy, order, and ndmin arguments to jax.numpy.array positionally is deprecated. Use keyword arguments instead. This matches the signature of numpy.array.
    • Python dict_values, generators, zip return type and iterators generally are deprecated by default when used as leaves in pytrees. In a future version of JAX, this will become an error, if you depend on using them as leaves, pass is_leaf to jax.tree.* methods.