Skip to content

Update jax requirement from <0.2.29,>=0.2.23 to >=0.2.23,<0.3.1

Vicentini Filippo requested to merge dependabot/pip/jax-gte-0.2.23-and-lt-0.3.1 into master

Created by: dependabot[bot]

Updates the requirements on jax to permit the latest version.

Release notes

Sourced from jax's releases.

Jax release v0.3.0

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

Sourced from jax's changelog.

jax 0.3.0 (Feb 10, 2022)

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)

Merge request reports