JAX v0.7.2
-
Breaking changes:
jax.dlpack.from_dlpackno longer accepts a DLPack capsule. This behavior was deprecated and is now removed. The function must be called with an array implementing__dlpack__and__dlpack_device__.
-
Changes
-
The minimum supported NumPy version is now 2.0. Since SciPy 1.13 is required for NumPy 2.0 support, the minimum supported SciPy version is now 1.13.
-
JAX now represents constants in its internal jaxpr representation as a
LiteralArray, which is a private JAX type that duck types as anumpy.ndarray. This type may be exposed to users viacustom_jvprules, for example, and may break code that usesisinstance(x, np.ndarray). If this breaks your code, you may convert these arrays to classic NumPy arrays usingnp.asarray(x).
-
-
Bug fixes
arr.view(dtype=None)now returns the array unchanged, matching NumPy's semantics. Previously it returned the array with a float dtype.