jax-ml/jax jax-v0.10.1

JAX v0.10.1

on GitHub

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.