Bump jaxlib from 0.1.61 to 0.4.4

[dependabot[bot]]

Bumps jaxlib from 0.1.61 to 0.4.4.

Release notes

Sourced from jaxlib's releases.

Jaxlib release v0.4.4

No release notes provided.

jaxlib release v0.4.3

No release notes provided.

jaxlib release v0.4.2

No release notes provided.

Jaxlib release v0.4.1

• Changes
  • Support for Python 3.7 has been dropped, in accordance with JAX's {ref}version-support-policy.
  • The behavior of XLA_PYTHON_CLIENT_MEM_FRACTION=.XX has been changed to allocate XX% of the total GPU memory instead of the previous behavior of using currently available GPU memory to calculate preallocation. Please refer to GPU memory allocation for more details.
  • The deprecated method .block_host_until_ready() has been removed. Use .block_until_ready() instead.

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

... (truncated)

Changelog

Sourced from jaxlib's changelog.

jax 0.4.4 (Feb 16, 2023)

• Changes
  • The implementation of jit and pjit has been merged. Merging jit and pjit changes the internals of JAX without affecting the public API of JAX. Before, jit was a final style primitive. Final style means that the creation of jaxpr was delayed as much as possible and transformations were stacked on top of each other. With the jit-pjit implementation merge, jit becomes an initial style primitive which means that we trace to jaxpr as early as possible. For more information see this section in autodidax. Moving to initial style should simplify JAX's internals and make development of features like dynamic shapes, etc easier. You can disable it only via the environment variable i.e. os.environ['JAX_JIT_PJIT_API_MERGE'] = '0'. The merge must be disabled via an environment variable since it affects JAX at import time so it needs to be disabled before jax is imported.
  • axis_resources argument of with_sharding_constraint is deprecated. Please use shardings instead. There is no change needed if you were using axis_resources as an arg. If you were using it as a kwarg, then please use shardings instead. axis_resources will be removed after 3 months from Feb 13, 2023.
  • added the {mod}jax.typing module, with tools for type annotations of JAX functions.
  • The following names have been deprecated:
    • jax.xla.Device and jax.interpreters.xla.Device: use jax.Device.
    • jax.experimental.maps.Mesh. Use jax.sharding.Mesh instead.
    • jax.experimental.pjit.NamedSharding: use jax.sharding.NamedSharding.
    • jax.experimental.pjit.PartitionSpec: use jax.sharding.PartitionSpec.
    • jax.interpreters.pxla.Mesh: use jax.sharding.Mesh.
    • jax.interpreters.pxla.PartitionSpec: use jax.sharding.PartitionSpec.
• Breaking Changes
  • the initial argument to reduction functions like :func:jax.numpy.sum is now required to be a scalar, consistent with the corresponding NumPy API. The previous behavior of broadcating the output against non-scalar initial values was an unintentional implementation detail ({jax-issue}[#14446](https://github.com/google/jax/issues/14446)).

jaxlib 0.4.4 (Feb 16, 2023)

• Breaking changes
  • Support for NVIDIA Kepler series GPUs has been removed from the default jaxlib builds. If Kepler support is needed, it is still possible to build jaxlib from source with Kepler support (via the --cuda_compute_capabilities=sm_35 option to build.py), however note that CUDA 12 has completely dropped support for Kepler GPUs.

jax 0.4.3 (Feb 8, 2023)

• Breaking changes
  • Deleted {func}jax.scipy.linalg.polar_unitary, which was a deprecated JAX extension to the scipy API. Use {func}jax.scipy.linalg.polar instead.

... (truncated)

Commits

• 58e46b4 Prepare for jax and jaxlib 0.4.4 release
• c6a99b6 Remove jax.interpreters.xla.lower_fun.
• a9e886f [jax2tf] Enable all native lowering jax2tf tests
• 454e4de [shape_poly] Fix the lowering for symbolic dimension expressions for division
• d0b42f2 Fix the simple bug on call_tf.replace_non_float and add unittest for floating...
• 26045c4 remove core.{aval_method,aval_property}
• d8514d0 Merge pull request #14500 from jakevdp:bcsr-matmul-test
• 0af9fff Replace uses of deprecated JAX sharding APIs with their new names in jax.shar...
• 1b2a318 remove core.axis_substitution_rules
• 768960b Fix pytype errors.
• 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)

[asmith26]

@dependabot ignore this dependency

[dependabot[bot]]

OK, I won't notify you about jaxlib again, unless you re-open this PR. 😢