Bump jaxlib from 0.1.75 to 0.3.0

[dependabot[bot]]

Bumps jaxlib from 0.1.75 to 0.3.0.

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).

Changelog

Sourced from jaxlib's changelog.

jax 0.3.0 (Feb 10, 2022)

• GitHub commits.

• Changes

  • jax version has been bumped to 0.3.0. Please see the design doc for the explanation.

jaxlib 0.3.0 (Feb 10, 2022)

• 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.

jax 0.2.28 (Feb 1, 2022)

• GitHub commits.
  • jax.jit(f).lower(...).compiler_ir() now defaults to the MHLO dialect if no dialect= is passed.
  • The jax.jit(f).lower(...).compiler_ir(dialect='mhlo') now returns an MLIR ir.Module object instead of its string representation.

jaxlib 0.1.76 (Jan 27, 2022)

• 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).

jax 0.2.27 (Jan 18 2022)

• GitHub commits.

• Breaking changes:

  • Support for NumPy 1.18 has been dropped, per the deprecation policy. Please upgrade to a supported NumPy version.
  • The host_callback primitives have been simplified to drop the special autodiff handling for hcb.id_tap and id_print. From now on, only the primals are tapped. The old behavior can be

... (truncated)

Commits

• 1ad3551 Release jax and jaxlib 0.3.0 as per the new release process.
• 4df01ee Merge pull request #9523 from google:yashk2810-patch-6
• ef11d53 Update TF commit for release
• d82bcc2 Add jax[_ci] option to account for the new release process.
• 0b135fb Merge pull request #9518 from LenaMartens:changelist/427732530
• 2ba8aec Checkify: Fix docstring formatting and polish enabled_errors sets.
• 1340fbb Strip named_shape and weak_type from aval when donating buffers.
• 744636d Merge pull request #9514 from google:yashk2810-patch-5
• cabc98c update TF commit for release
• 8af0d8d Add complex number DLPack support to JAX and TensorFlow.
• 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)

[codecov[bot]]

Codecov Report

Merging #55 (b947492) into main (70b7e26) will not change coverage. The diff coverage is n/a.

@@           Coverage Diff           @@
##             main      #55   +/-   ##
=======================================
  Coverage   49.78%   49.78%           
=======================================
  Files          10       10           
  Lines        1418     1418           
=======================================
  Hits          706      706           
  Misses        712      712           

---

Continue to review full report at Codecov.

Legend - Click here to learn more Δ = absolute <relative> (impact), ø = not affected, ? = missing data Powered by Codecov. Last update 70b7e26...b947492. Read the comment docs.

[dependabot[bot]]

Looks like jaxlib is no longer a dependency, so this is no longer needed.