Bump jaxlib from 0.1.56 to 0.3.25

[dependabot]

Bumps jaxlib from 0.1.56 to 0.3.25.

Release notes

Sourced from jaxlib's releases.

Jaxlib release v0.3.25

No release notes provided.

Jaxlib release v0.3.24

No release notes provided.

jaxlib-v0.3.22

No release notes provided.

jaxlib release v0.3.20

Notable changes:

• Fixes support for limiting the visible CUDA devices viajax_cuda_visible_devices in distributed jobs. This functionality is needed for the JAX/SLURM integration on GPU (#12533).

jaxlib version 0.3.15

No release notes provided.

Jaxlib release v0.3.14

No release notes provided.

Jaxlib release v0.3.10

No release notes provided.

Jaxlib v0.3.7

• Linux wheels are now built conforming to the manylinux2014 standard, instead of manylinux2010.

Jaxlib release v0.3.0

• Changes
  • Bazel 5.0.0 is now required to build jaxlib.
  • jaxlib version has been bumped to 0.3.0. Please see the design doc for the explanation.

Jaxlib v0.1.76

• New features
  • Includes precompiled SASS for NVidia compute capability 8.0 GPUS (e.g. A100). Removes precompiled SASS for compute capability 6.1 so as not to increase the number of compute capabilities: GPUs with compute capability 6.1 can use the 6.0 SASS.
  • With jaxlib 0.1.76, JAX uses the MHLO MLIR dialect as its primary target compiler IR by default.
• Breaking changes
  • Support for NumPy 1.18 has been dropped, per the deprecation policy. Please upgrade to a supported NumPy version.
• Bug fixes
  • Fixed a bug where apparently identical pytreedef objects constructed by different routes do not compare as equal (#9066).
  • The JAX jit cache requires two static arguments to have identical types for a cache hit (#9311).

Jaxlib release v0.1.75

• New features:
  • Support for python 3.10.

... (truncated)

Changelog

Sourced from jaxlib's changelog.

jax 0.3.25 (Nov 15, 2022)

• Changes
  • {func}jax.numpy.linalg.pinv now supports the hermitian option.
  • {func}jax.scipy.linalg.hessenberg is now supported on CPU only. Requires jaxlib > 0.3.24.
  • New functions {func}jax.lax.linalg.hessenberg, {func}jax.lax.linalg.tridiagonal, and {func}jax.lax.linalg.householder_product were added. Householder reduction is currently CPU-only and tridiagonal reductions are supported on CPU and GPU only.
  • The gradients of svd and jax.numpy.linalg.pinv are now computed more economically for non-square matrices.
• Breaking Changes
  • Deleted the jax_experimental_name_stack config option.
  • Convert a string axis_names arguments to the {class}jax.experimental.maps.Mesh constructor into a singleton tuple instead of unpacking the string into a sequence of character axis names.

jaxlib 0.3.25 (Nov 15, 2022)

• Changes
  • Added support for tridiagonal reductions on CPU and GPU.
  • Added support for upper Hessenberg reductions on CPU.
• Bugs
  • Fixed a bug that meant that frames in tracebacks captured by JAX were incorrectly mapped to source lines under Python 3.10+

jax 0.3.24 (Nov 4, 2022)

• Changes
  • JAX should be faster to import. We now import scipy lazily, which accounted for a significant fraction of JAX's import time.
  • Setting the env var JAX_PERSISTENT_CACHE_MIN_COMPILE_TIME_SECS=$N can be used to limit the number of cache entries written to the persistent cache. By default, computations that take 1 second or more to compile will be cached.
    • Added {func}jax.scipy.stats.mode.
  • The default device order used by pmap on TPU if no order is specified now matches jax.devices() for single-process jobs. Previously the two orderings differed, which could lead to unnecessary copies or out-of-memory errors. Requiring the orderings to agree simplifies matters.
• Breaking Changes
  • {func}jax.numpy.gradient now behaves like most other functions in {mod}jax.numpy, and forbids passing lists or tuples in place of arrays ({jax-issue}[#12958](https://github.com/google/jax/issues/12958))
  • Functions in {mod}jax.numpy.linalg and {mod}jax.numpy.fft now uniformly require inputs to be array-like: i.e. lists and tuples cannot be used in place of arrays. Part of {jax-issue}[#7737](https://github.com/google/jax/issues/7737).
• Deprecations
  • jax.sharding.MeshPspecSharding has been renamed to jax.sharding.NamedSharding. jax.sharding.MeshPspecSharding name will be removed in 3 months.

jaxlib 0.3.24 (Nov 4, 2022)

... (truncated)

Commits

• a683186 Use the 11/09 libtpu build for jaxlib release since that passes all the tests.
• 7635df8 Remove custom potrf kernels in favor of native XLA cholesky support.
• f36084a Update the values for jaxlib release (again)
• 0da02dd Update the values needed for a jaxlib release
• 465d4ec Merge pull request #13243 from google:yashk2810-patch-19
• 50f7512 Update WORKSPACE for jaxlib release
• e6c4d4a Add docstrings for Sharding classes. Right now I am only documenting `Shard...
• c42bad8 Make MeshPspecSharding an alias for NamedSharding (it was the other way a...
• d1e26d9 Merge pull request #13139 from mattjj:djax-vmap4
• f715ca6 Merge pull request #13236 from skye:build_chat
• 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 merge will merge this PR after your CI passes on it
• @dependabot squash and merge will squash and merge this PR after your CI passes on it
• @dependabot cancel merge will cancel a previously requested merge and block automerging
• @dependabot reopen will reopen this PR if it is closed
• @dependabot close will close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually
• @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)

[dependabot]

Superseded by #403.