JAX v0.10.1

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.