●Change8

Migrating to JAX jax-v0.10.1

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

Released: 5/20/2026

1
Breaking Changes
1
Migration Steps
13
Affected Symbols

⚠️ Check Your Code

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

jax.image.resizejax.scipy.linalg.hadamardjax.scipy.linalg.circulantjax.scipy.linalg.dftjax.scipy.linalg.lesliejax.scipy.linalg.companionjax.scipy.linalg.fiedlerjax.scipy.linalg.helmertjax.random.key_dtypejax.random.keywrap_key_datajax.numpy.arrayjax.tree.*

Breaking Changes

●Issue #1

The `with mesh:` context manager is deprecated. Users must switch to using `with jax.set_mesh(mesh):` instead.

Migration Steps

  1. 1
    Replace usage of the deprecated `with mesh:` context manager with `with jax.set_mesh(mesh):`.

Release Summary

This release introduces several new linear algebra functions in `jax.scipy.linalg` and updates RNG API handling by moving related functionality to dtypes. It also deprecates positional arguments for certain array creation parameters and modifies the mesh context manager usage.

Need More Details?

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

View Full Changelog