[air](deps): Bump jax from 0.4.33 to 0.10.2 in /python/requirements/ml

[dependabot]

Bumps jax from 0.4.33 to 0.10.2.

Release notes

Sourced from jax's releases.

JAX v0.10.2

• New features
  • Added jax.scipy.linalg.invhilbert for the closed-form inverse of the Hilbert matrix ({jax-issue}[#10144](https://github.com/jax-ml/jax/issues/10144)).
  • Added jax.scipy.linalg.invpascal for the inverse of the Pascal matrix ([#10144](https://github.com/jax-ml/jax/issues/10144)).
  • Added jax.scipy.linalg.fiedler_companion for constructing the pentadiagonal Fiedler companion matrix of a polynomial ([#10144](https://github.com/jax-ml/jax/issues/10144)).
  • Added jax.ShapeDtypeStruct.like -- a shortcut for constructing a jax.ShapeDtypeStruct from an object with shape and dtype attributes.
• We have made a number of improvements to JAX tracing performance.

JAX v0.10.1

• 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.

... (truncated)

Changelog

Sourced from jax's changelog.

JAX 0.10.2 (June 17, 2026)

• New features
  • Added {func}jax.scipy.linalg.invhilbert for the closed-form inverse of the Hilbert matrix ({jax-issue}[#10144](https://github.com/jax-ml/jax/issues/10144)).
  • Added {func}jax.scipy.linalg.invpascal for the inverse of the Pascal matrix ({jax-issue}[#10144](https://github.com/jax-ml/jax/issues/10144)).
  • Added {func}jax.scipy.linalg.fiedler_companion for constructing the pentadiagonal Fiedler companion matrix of a polynomial ({jax-issue}[#10144](https://github.com/jax-ml/jax/issues/10144)).
  • Added {func}jax.ShapeDtypeStruct.like -- a shortcut for constructing a {class}jax.ShapeDtypeStruct from an object with shape and dtype attributes.

JAX 0.10.1 (May 20, 2026)

• New features

  • Added ResizeMethod.AREA to {func}jax.image.resize, which matches TensorFlow's AREA resizing ({jax-issue}[#20098](https://github.com/jax-ml/jax/issues/20098)).
  • Added {func}jax.scipy.linalg.hadamard for constructing Hadamard matrices ({jax-issue}[#10144](https://github.com/jax-ml/jax/issues/10144)).
  • Added {func}jax.scipy.linalg.circulant for constructing circulant matrices ({jax-issue}[#10144](https://github.com/jax-ml/jax/issues/10144)).
  • Added {func}jax.scipy.linalg.dft for constructing discrete Fourier transform matrices ({jax-issue}[#10144](https://github.com/jax-ml/jax/issues/10144)).
  • Added {func}jax.scipy.linalg.leslie for constructing Leslie matrices ({jax-issue}[#10144](https://github.com/jax-ml/jax/issues/10144)).
  • Added {func}jax.scipy.linalg.companion for constructing companion matrices from polynomial coefficients ({jax-issue}[#10144](https://github.com/jax-ml/jax/issues/10144)).
  • Added {func}jax.scipy.linalg.fiedler for constructing symmetric Fiedler matrices ({jax-issue}[#10144](https://github.com/jax-ml/jax/issues/10144)).
  • Added {func}jax.scipy.linalg.helmert for constructing Helmert matrices ({jax-issue}[#10144](https://github.com/jax-ml/jax/issues/10144)).
  • Added {func}jax.scipy.special.boxcox and {func}jax.scipy.special.boxcox1p for the Box-Cox power transformation.
  • Moved RNG APIs from "implementations" to dtypes ({jax-issue}[#27854](https://github.com/jax-ml/jax/issues/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 {func}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

... (truncated)

Commits

• 990e6a0 Disable gpu_pallas_distributed_nvshmem_test_gpu which was failing at the
• 4437b23 Synchronize PGLE profiler tests before stopping traces
• 3815e4e Add a workaround for linux aarch64 python 3.14t pytest builds.
• 272e378 Refactor Windows Docker GCE metadata access
• 310278e Prepare for JAX release 0.10.2
• 7db3a70 [Pallas:MGPU] Add support for batch dimensions to async_store_tmem.
• 5597e90 [Mosaic:GPU] Disable set_xla_gpu_enable_allocator_spatial_partitioning for nv...
• 728e4bf Add functionality to blanket disable the custom call safety checks in export.
• 26113fa Merge pull request #38432 from vfdev-5:pallas-emit-pipeline-raises-error-empt...
• c15bacc Pallas emit_pipeline raises ValueError on empty grid
• Additional commits viewable in compare view

Dependabot will resolve any conflicts with this PR as long as you don't alter it yourself. You can also trigger a rebase manually by commenting @dependabot rebase.

---

Dependabot commands and options

You can trigger Dependabot actions by commenting on this PR:

• @dependabot rebase will rebase this PR
• @dependabot recreate will recreate this PR, overwriting any edits that have been made to it
• @dependabot show <dependency name> ignore conditions will show all of the ignore conditions of the specified dependency
• @dependabot ignore this major version will close this PR and stop Dependabot creating any more for this major version (unless you reopen the PR or upgrade to it yourself)
• @dependabot ignore this minor version will close this PR and stop Dependabot creating any more for this minor version (unless you reopen the PR or upgrade to it yourself)
• @dependabot ignore this dependency will close this PR and stop Dependabot creating any more for this dependency (unless you reopen the PR or upgrade to it yourself)