Bump jaxlib from 0.1.61 to 0.1.74

[dependabot[bot]]

Bumps jaxlib from 0.1.61 to 0.1.74.

Changelog

Sourced from jaxlib's changelog.

jaxlib 0.1.74 (Unreleased)

• Enabled peer-to-peer copies between GPUs. Previously, GPU copies were bounced via the host, which is usually slower.
• Added experimental MLIR Python bindings for use by JAX.

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
    

... (truncated)

Commits

• b53a174 Merge pull request #8564 from hawkinsp:testfix
• 9e09b51 Merge pull request #8381 from LenaMartens:changelist/405399581
• 6883571 Merge pull request #8561 from mattjj:add-donated-invars-to-xlacomputation
• e14fea3 Overload jnp ops which are polymorphic to an array's value and support PRNGKeys.
• 1bcedd5 Fix test failures and update changelog.
• 5d35b8a add donated_invars to xla.XlaComputation
• b7e3129 Merge pull request #8562 from hawkinsp:jaxlib
• a864c95 Update XLA.
• 9491414 Merge pull request #8499 from jakevdp:load-wrapper
• 1137aa1 Properly handle bfloat16 in jnp.load()
• 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 #244.