Bump jaxlib from 0.1.56 to 0.1.72

[dependabot]

Bumps jaxlib from 0.1.56 to 0.1.72.

Changelog

Sourced from jaxlib's changelog.

jaxlib 0.1.72 (Oct 12, 2021)

• Breaking changes:
  • Support for CUDA 10.2 and CUDA 10.1 has been dropped. Jaxlib now supports CUDA 11.1+.

jax 0.2.21 (Sept 23, 2021)

• GitHub commits.
• Breaking Changes
  • jax.api has been removed. Functions that were available as jax.api.* were aliases for functions in jax.*; please use the functions in jax.* instead.
  • jax.partial, and jax.lax.partial were accidental exports that have now been removed. Use functools.partial from the Python standard library instead.
  • Boolean scalar indices now raise a TypeError; previously this silently returned wrong results ({jax-issue}[#7925](https://github.com/google/jax/issues/7925)).
  • Many more jax.numpy functions now require array-like inputs, and will error if passed a list ({jax-issue}[#7747](https://github.com/google/jax/issues/7747) {jax-issue}[#7802](https://github.com/google/jax/issues/7802) {jax-issue}[#7907](https://github.com/google/jax/issues/7907)). See {jax-issue}[#7737](https://github.com/google/jax/issues/7737) for a discussion of the rationale behind this change.
  • When inside a transformation such as jax.jit, jax.numpy.array always stages the array it produces into the traced computation. Previously jax.numpy.array would sometimes produce a on-device array, even under a jax.jit decorator. This change may break code that used JAX arrays to perform shape or index computations that must be known statically; the workaround is to perform such computations using classic NumPy arrays instead.
  • jnp.ndarray is now a true base-class for JAX arrays. In particular, this means that for a standard numpy array x, isinstance(x, jnp.ndarray) will now return False ({jax-issue}7927).
• New features:
  • Added {func}jax.numpy.insert implementation ({jax-issue}[#7936](https://github.com/google/jax/issues/7936)).

jax 0.2.20 (Sept 2, 2021)

• GitHub commits.
• Breaking Changes
  • jnp.poly* functions now require array-like inputs ({jax-issue}[#7732](https://github.com/google/jax/issues/7732))
  • jnp.unique and other set-like operations now require array-like inputs ({jax-issue}[#7662](https://github.com/google/jax/issues/7662))

jaxlib 0.1.71 (Sep 1, 2021)

• Breaking changes:
  • Support for CUDA 11.0 and CUDA 10.1 has been dropped. Jaxlib now supports CUDA 10.2 and CUDA 11.1+.

jax 0.2.19 (Aug 12, 2021)

• GitHub commits.
• Breaking changes:

... (truncated)

Commits

• d8b7bd5 Merge pull request #8181 from skye:workspace
• 49a3b37 Update WORKSPACE for jaxlib 0.1.72 release
• ef3e3b5 Rollback fix repr() of jit-compiled functions
• 7fa6b1b Merge pull request #8160 from Jakob-Unfried:main
• 40ee7f2 Merge pull request #8164 from jakevdp:bcoo-dedupe-pad
• ccc8b72 Merge pull request #8153 from apaszke:while-loop-batching
• bec943c fix fori_loop and scan when trivial and with disable_jit
• a8ce40b Merge pull request #7989 from gnecula:remat_docstring
• b8dc8ca Fix while loop batching rule for loops with constant bodies
• cfa0f78 Merge pull request #8140 from jakevdp:nanstd-grad
• 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]

Superseded by #249.