Bump jaxlib from 0.1.61 to 0.3.7

[dependabot[bot]]

Bumps jaxlib from 0.1.61 to 0.3.7.

Release notes

Sourced from jaxlib's releases.

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.

Jaxlib release v0.1.74

No release notes provided.

Jaxlib release v0.1.73

No release notes provided.

jaxlib release v0.1.72

No release notes provided.

Jaxlib release v0.1.71

No release notes provided.

jaxlib v0.1.70 release

No release notes provided.

Changelog

Sourced from jaxlib's changelog.

jax 0.3.7 (April 15, 2022)

• GitHub commits.
• Changes:
  • Fixed a performance problem if the indices passed to {func}jax.numpy.take_along_axis were broadcasted ({jax-issue}[#10281](https://github.com/google/jax/issues/10281)).
  • {func}jax.scipy.special.expit and {func}jax.scipy.special.logit now require their arguments to be scalars or JAX arrays. They also now promote integer arguments to floating point.
  • The DeviceArray.tile() method is deprecated, because numpy arrays do not have a tile() method. As a replacement for this, use {func}jax.numpy.tile ({jax-issue}[#10266](https://github.com/google/jax/issues/10266)).

jaxlib 0.3.7 (April 15, 2022)

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

jax 0.3.6 (April 12, 2022)

• GitHub commits.
• Changes:
  • Upgraded libtpu wheel to a version that fixes a hang when initializing a TPU pod. Fixes #10218.
• Deprecations:
  • {mod}jax.experimental.loops is being deprecated. See {jax-issue}[#10278](https://github.com/google/jax/issues/10278) for an alternative API.

jax 0.3.5 (April 7, 2022)

• GitHub commits.
• Changes:
  • added {func}jax.random.loggamma & improved behavior of {func}jax.random.beta and {func}jax.random.dirichlet for small parameter values ({jax-issue}[#9906](https://github.com/google/jax/issues/9906)).
  • the private lax_numpy submodule is no longer exposed in the jax.numpy namespace ({jax-issue}[#10029](https://github.com/google/jax/issues/10029)).
  • added array creation routines {func}jax.numpy.frombuffer, {func}jax.numpy.fromfunction, and {func}jax.numpy.fromstring ({jax-issue}[#10049](https://github.com/google/jax/issues/10049)).
  • DeviceArray.copy() now returns a DeviceArray rather than a np.ndarray ({jax-issue}[#10069](https://github.com/google/jax/issues/10069))
  • added {func}jax.scipy.linalg.rsf2csf
• Deprecations:
  • {func}jax.nn.normalize is being deprecated. Use {func}jax.nn.standardize instead ({jax-issue}[#9899](https://github.com/google/jax/issues/9899)).
  • {func}jax.tree_util.tree_multimap is deprecated. Use {func}jax.tree_util.tree_map instead ({jax-issue}[#5746](https://github.com/google/jax/issues/5746)).
  • jax.experimental.sharded_jit is deprecated. Use pjit instead.

jaxlib 0.3.5 (April 7, 2022)

• Bug fixes
  • Fixed a bug where double-precision complex-to-real IRFFTs would mutate their input buffers on GPU ({jax-issue}[#9946](https://github.com/google/jax/issues/9946)).
  • Fixed incorrect constant-folding of complex scatters ({jax-issue}[#10159](https://github.com/google/jax/issues/10159))

... (truncated)

Commits

• 1189c3c Merge pull request #10312 from hawkinsp:jaxlib
• 6b33c55 2nd attempt at jax/jaxlib 0.3.7 release.
• a303e4b Merge pull request #10266 from jakevdp:ndarray-tile
• 6eec758 [JAX] Don't pass non-integer indices to jnp.take_along_axis
• 7ffdac0 Skip remote_transfer_test on cloud TPU.
• a4b8a44 Merge pull request #10288 from YouJiacheng:patch-7
• be5c84d Deprecate DeviceArray.tile method
• 4ff6b1f Fix PRNGKeyArray.broadcast_to with scalar shape
• 470f58c Merge pull request #10309 from hawkinsp:jaxlib
• 52a97f2 Jax 0.3.7 and jaxlib 0.3.7 release.
• 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[bot]]

Superseded by #296.