Bump jaxlib from 0.1.56 to 0.1.58

[dependabot]

Bumps jaxlib from 0.1.56 to 0.1.58.

Changelog

Sourced from jaxlib's changelog.

jaxlib 0.1.58 (Unreleased)

• Fixed a bug that meant JAX sometimes return platform-specific types (e.g., np.cint) instead of standard types (e.g., np.int32). (#4903)
• Fixed a crash when constant-folding certain int16 operations. (#4971)
• Added an is_leaf predicate to pytree.flatten.

jaxlib 0.1.57 (November 12 2020)

• Fixed manylinux2010 compliance issues in GPU wheels.
• Switched the CPU FFT implementation from Eigen to PocketFFT.
• Fixed a bug where the hash of bfloat16 values was not correctly initialized and could change (#4651).
• Add support for retaining ownership when passing arrays to DLPack (#4636).
• Fixed a bug for batched triangular solves with sizes greater than 128 but not a multiple of 128.
• Fixed a bug when performing concurrent FFTs on multiple GPUs (#3518).
• Fixed a bug in profiler where tools are missing (#4427).
• Dropped support for CUDA 10.0.

jax 0.2.5 (October 27 2020)

• GitHub commits <https://github.com/google/jax/compare/jax-v0.2.4...jax-v0.2.5>_.

• Improvements:

  • Ensure that check_jaxpr does not perform FLOPS. See [#4650](https://github.com/google/jax/issues/4650) <https://github.com/google/jax/pull/4650>_.
  • Expanded the set of JAX primitives converted by jax2tf. See primitives_with_limited_support.md <https://github.com/google/jax/blob/master/jax/experimental/jax2tf/primitives_with_limited_support.md>_.

jax 0.2.4 (October 19 2020)

• GitHub commits <https://github.com/google/jax/compare/jax-v0.2.3...jax-v0.2.4>_.

• Improvements:

  • Add support for remat to jax.experimental.host_callback. See [#4608](https://github.com/google/jax/issues/4608) <https://github.com/google/jax/pull/4608>_.

• Deprecations

  • Indexing with non-tuple sequences is now deprecated, following a similar deprecation in Numpy. In a future release, this will result in a TypeError. See [#4564](https://github.com/google/jax/issues/4564) <https://github.com/google/jax/pull/4564>_.

Commits

• 21551c2 Bump JAX version to 0.1.58.
• 82d6c6c Added better error messages. (#2058)
• e0ed5ad Allow JVP for SVD when not computing singular vectors (#2076)
• 832bb71 Add missing BUILD dependency. (#2089)
• ffa198e Fix test failure on TPU. (#2088)
• a15aa9b include call stack + transforms in XLA metadata (#2073)
• 0fca476 Implement np.linalg.matrix_rank (#2008)
• 632326a Add unsupported wrapper around XLA RngUniform API. (#2068)
• 0c95c26 Implement np.linalg.matrix_power (#2042)
• f1339cd Remove missing PPA in Dockerfile. (#2061)
• 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 #160.