Bump jax from 0.10.1 to 0.10.2

[dependabot]

Bumps jax from 0.10.1 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.

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.

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

You can trigger a rebase of this PR 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)

Note
Automatic rebases have been disabled on this pull request as it has been open for over 30 days.