Migrating to JAX jax-v0.6.2
Version jax-v0.6.2 introduces 1 breaking change. This guide details how to update your code.
Released: 6/17/2025
1
Breaking Changes
0
Migration Steps
1
Affected Symbols
⚠️ Check Your Code
If you use any of these symbols, you need to read this guide:
jax.tree.broadcastBreaking Changes
●Issue #1
The library now requires NumPy >=1.26 and SciPy >=1.12; environments using older versions must upgrade those dependencies.
Release Summary
This release introduces the new `jax.tree.broadcast` helper and raises the minimum required versions of NumPy and SciPy.
Need More Details?
View the full release notes and all changes for JAX jax-v0.6.2.
View Full Changelog