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.