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

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)