●Change8

Migrating to JAX jax-v0.7.2

Version jax-v0.7.2 introduces 1 breaking change. This guide details how to update your code.

Released: 9/16/2025

1
Breaking Changes
4
Migration Steps
5
Affected Symbols

⚠️ Check Your Code

If you use any of these symbols, you need to read this guide:

jax.dlpack.from_dlpackjax2tf.convertjax_pmap_no_rank_reductionjax.random.randintLiteralArray

Breaking Changes

●Issue #1

jax.dlpack.from_dlpack no longer accepts a DLPack capsule; it must be called with an array implementing __dlpack__ and __dlpack_device__.

Migration Steps

  1. 1
    Update calls to jax.dlpack.from_dlpack to pass an object that implements __dlpack__ and __dlpack_device__ instead of a raw DLPack capsule.
  2. 2
    If code relies on isinstance(x, np.ndarray) for JAX constants, convert the value with np.asarray(x) to obtain a classic NumPy array.
  3. 3
    Remove usage of the enable_xla and native_serialization arguments from jax2tf.convert calls.
  4. 4
    Stop setting jax_pmap_no_rank_reduction to False; rely on the default True behavior.

Release Summary

JAX drops support for raw DLPack capsules in jax.dlpack.from_dlpack, raises minimum NumPy/SciPy versions, and introduces several deprecations and bug fixes.

Need More Details?

View the full release notes and all changes for JAX jax-v0.7.2.

View Full Changelog