Skip to content

Update jax requirement from <0.2.25,>=0.2.21 to >=0.2.21,<0.2.26

Vicentini Filippo requested to merge dependabot/pip/jax-gte-0.2.21-and-lt-0.2.26 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.2.25

  • New features:

    • (Experimental) jax.distributed.initialize exposes multi-host GPU backend.
    • jax.random.permutation supports new independent keyword argument ({jax-issue}[#8430](https://github.com/google/jax/issues/8430))

  • Breaking changes

    • Moved jax.experimental.stax to jax.example_libraries.stax
    • Moved jax.experimental.optimizers to jax.example_libraries.optimizers

  • New features:

    • Added jax.lax.linalg.qdwh.
Changelog

Sourced from jax's changelog.

jax 0.2.25 (Nov 10, 2021)

  • GitHub commits.

  • New features:

    • (Experimental) jax.distributed.initialize exposes multi-host GPU backend.
    • jax.random.permutation supports new independent keyword argument ({jax-issue}[#8430](https://github.com/google/jax/issues/8430))

  • Breaking changes

    • Moved jax.experimental.stax to jax.example_libraries.stax
    • Moved jax.experimental.optimizers to jax.example_libraries.optimizers

  • New features:

    • Added jax.lax.linalg.qdwh.

jax 0.2.24 (Oct 19, 2021)

  • GitHub commits.

  • New features:

    • jax.random.choice and jax.random.permutation now support multidimensional arrays and an optional axis argument ({jax-issue}[#8158](https://github.com/google/jax/issues/8158))

  • Breaking changes:

    • jax.numpy.take and jax.numpy.take_along_axis now require array-like inputs (see {jax-issue}[#7737](https://github.com/google/jax/issues/7737))

jaxlib 0.1.73 (Oct 18, 2021)

  • Multiple cuDNN versions are now supported for jaxlib GPU cuda11 wheels.

    • cuDNN 8.2 or newer. We recommend using the cuDNN 8.2 wheel if your cuDNN installation is new enough, since it supports additional functionality.
    • cuDNN 8.0.5 or newer.

  • Breaking changes:

    • The install commands for GPU jaxlib are as follows:

      pip install --upgrade pip
Installs the wheel compatible with CUDA 11 and cuDNN 8.2 or newer.
pip install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_releases.html
Installs the wheel compatible with Cuda 11 and cudnn 8.2 or newer.
pip install jax[cuda11_cudnn82] -f https://storage.googleapis.com/jax-releases/jax_releases.html
Installs the wheel compatible with Cuda 11 and cudnn 8.0.5 or newer.
pip install jax[cuda11_cudnn805] -f https://storage.googleapis.com/jax-releases/jax_releases.html

jax 0.2.22 (Oct 12, 2021)

... (truncated)

Commits
  • 821fcaa Make the pjit docs clear about who does local and global communication
  • 0f47712 Implements a lowering function for tf.expm1 and adds tests.
  • 66009d6 Merge pull request #8331 from jakevdp:odeint-args
  • beb473d Merge pull request #8320 from jakevdp:hist-density
  • ca4e430 Merge pull request #8332 from jakevdp:bcoo-block
  • 3338947 odeint: args validation allows pytrees
  • 11127ad [JAX] Replace users of deprecated jax.lib.xla_bridge functions with more publ...
  • 20062e5 [sparse] add .block_until_ready() to sparse objects
  • 7a5e843 Internal change
  • bc1c6b1 [JAX] Fix sphinx formatting issue
  • 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