Bump jax from 0.11.1 to 0.11.2

[dependabot]

Bumps jax from 0.11.1 to 0.11.2.

Release notes

Sourced from jax's releases.

JAX v0.11.2

• New features
  • Added jax.numpy.minmax (and jnp.minmax), which returns both the minimum and maximum of an array, matching NumPy 2.3+ (#40089).
  • Added jax.lax.log2 and primitive jax.lax.log2_p, making log2 a first-class primitive in JAX (jax.numpy.log2 now lowers via jax.lax.log2).
  • Added jax.lax.one_minus_square primitive to accurately compute 1 - x^2 near $\pm 1$ and provide accurate derivatives near $0$.
  • Added jax.export.symbolic_dim_bounds for querying conservative bounds on symbolic dimension expressions (#40006).
  • Added frozendict support to JAX pytrees for Python 3.15 (PEP 814).
  • jax.distributed.initialize can now secure the coordination service with mutual TLS via the new mtls_cert_file, mtls_key_file, mtls_ca_file, mtls_peer_uri_prefix and verify_secure_credentials arguments (or the JAX_MTLS_CERT_FILE, JAX_MTLS_KEY_FILE, JAX_MTLS_CA_FILE, JAX_MTLS_PEER_URI_PREFIX and JAX_DISTRIBUTED_VERIFY_SECURE_CREDENTIALS environment variables).
  • Added cluster detection support for Open MPI 5 in jax.distributed.initialize (#40512).
  • Added support for reading TPU_PROCESS_ADDRESSES_PATH in GKE TPU clusters.
  • Widened jax.random.generalized_normal's p parameter type from float to RealArray, allowing array-valued shape parameters (#40126).
  • Added exclude_argnames argument to jax.experimental.program_order.
  • [oneAPI GPU] Added oneMKL LAPACK solver kernels for QR decomposition (geqrf, orgqr/ungqr, ormqr/unmqr), LU decomposition (getrf), symmetric/Hermitian eigenvalue decomposition (syevd/heevd), SVD (gesvd), and hybrid solver kernels (geqp3, eig) (#40000, #40186, #40543).
  • jaxlib wheels now ship C++ FFI extension headers (collectives.h, record.h) to support out-of-tree plugins (#40333).
• Breaking changes
  • Removed deprecated jax.experimental.shard_alike. Use explicit sharding mode instead (see sharding).
• Changes
  • Optimized large JAX jax.sharding.Mesh construction by avoiding redundant device array allocations and copies.
  • Thread-safety improvements across JAX and jaxlib for free-threaded Python (Python 3.13t, 3.14t, 3.15t).
  • inline=True in jax.jit now corresponds to jax.Inline.JAX_LATE instead of jax.Inline.JAX_EARLY.
  • The minimum CuDNN version for CUDA 12 is v9.10.2.
  • JAX now uses Bazel 8.7.0 to build from source.
  • JAX now uses Bzlmod for its Bazel builds instead of WORKSPACE.
  • On GPU, multi-dimensional inverse real FFTs (jax.numpy.fft.irfftn, jax.numpy.fft.irfft2 and jax.lax.fft with FftType.IRFFT) are again lowered to a single C2R transform, as before JAX 0.10.0, instead of an IFFT over the outer axes and a 1-D IRFFT with two transposes. The input is first made Hermitian-symmetric along the outer axes, which does not change the result under NumPy's convention (only the last axis is

... (truncated)

Changelog

Sourced from jax's changelog.

JAX 0.11.2 (September 17, 2026)

• New features

  • Added {func}jax.numpy.minmax (and jnp.minmax), which returns both the minimum and maximum of an array, matching NumPy 2.3+ ({jax-issue}[#40089](https://github.com/jax-ml/jax/issues/40089)).
  • Added {func}jax.lax.log2 and primitive {data}jax.lax.log2_p, making log2 a first-class primitive in JAX ({func}jax.numpy.log2 now lowers via jax.lax.log2).
  • Added {func}jax.lax.one_minus_square primitive to accurately compute 1 - x^2 near $\pm 1$ and provide accurate derivatives near $0$.
  • Added {func}jax.export.symbolic_dim_bounds for querying conservative bounds on symbolic dimension expressions ({jax-issue}[#40006](https://github.com/jax-ml/jax/issues/40006)).
  • Added {class}frozendict support to JAX pytrees for Python 3.15 (PEP 814).
  • {func}jax.distributed.initialize can now secure the coordination service with mutual TLS via the new mtls_cert_file, mtls_key_file, mtls_ca_file, mtls_peer_uri_prefix and verify_secure_credentials arguments (or the JAX_MTLS_CERT_FILE, JAX_MTLS_KEY_FILE, JAX_MTLS_CA_FILE, JAX_MTLS_PEER_URI_PREFIX and JAX_DISTRIBUTED_VERIFY_SECURE_CREDENTIALS environment variables).
  • Added cluster detection support for Open MPI 5 in {func}jax.distributed.initialize ({jax-issue}[#40512](https://github.com/jax-ml/jax/issues/40512)).
  • Added support for reading TPU_PROCESS_ADDRESSES_PATH in GKE TPU clusters.
  • Widened {func}jax.random.generalized_normal's p parameter type from float to RealArray, allowing array-valued shape parameters ({jax-issue}[#40126](https://github.com/jax-ml/jax/issues/40126)).
  • Added exclude_argnames argument to {func}jax.experimental.program_order.
  • [oneAPI GPU] Added oneMKL LAPACK solver kernels for QR decomposition (geqrf, orgqr/ungqr, ormqr/unmqr), LU decomposition (getrf), symmetric/Hermitian eigenvalue decomposition (syevd/heevd), SVD (gesvd), and hybrid solver kernels (geqp3, eig) ({jax-issue}[#40000](https://github.com/jax-ml/jax/issues/40000), {jax-issue}[#40186](https://github.com/jax-ml/jax/issues/40186), {jax-issue}[#40543](https://github.com/jax-ml/jax/issues/40543)).
  • jaxlib wheels now ship C++ FFI extension headers (collectives.h, record.h) to support out-of-tree plugins ({jax-issue}[#40333](https://github.com/jax-ml/jax/issues/40333)).

• Breaking changes

  • Removed deprecated jax.experimental.shard_alike. Use explicit sharding mode instead (see {ref}jax-201-sharding).

• Changes

  • Optimized large JAX {class}jax.sharding.Mesh construction by avoiding redundant device array allocations and copies.
  • Thread-safety improvements across JAX and jaxlib for free-threaded Python (Python 3.13t, 3.14t, 3.15t).
  • inline=True in {func}jax.jit now corresponds to {attr}jax.Inline.JAX_LATE instead of {attr}jax.Inline.JAX_EARLY.
  • The minimum CuDNN version for CUDA 12 is v9.10.2.
  • JAX now uses Bazel 8.7.0 to build from source.
  • JAX now uses Bzlmod for its Bazel builds instead of WORKSPACE.
  • On GPU, multi-dimensional inverse real FFTs ({func}jax.numpy.fft.irfftn, {func}jax.numpy.fft.irfft2 and {func}jax.lax.fft with FftType.IRFFT) are again lowered to a single C2R transform, as before JAX 0.10.0, instead

... (truncated)

Commits

• 3254480 Prepare for JAX release 0.11.2
• b59d695 Fix testSincValuesAndDerivativesLargeMagnitude with older NumPy versions.
• a532f8e Optimize large JAX Mesh construction
• 0208628 Merge pull request #40752 from hawkinsp:libtpu
• 9310faa Skip LayoutTest.test_host_auto_layout on CPU and GPU.
• f907e2a Bump libtpu pin to v0.0.48.
• 73b3c04 Merge pull request #40693 from Intel-tensorflow:mraunak/mkl_def_toolchain_update
• 5672063 Update XLA dependency to use revision http://github.com/openxla/xla/commit/91...
• 171383a Support JAX AUTO layout for host memory space in XLA.
• 9a863b3 Keep all non-manual dims open in partial manual shard_map lowering
• 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)