Bump jaxlib from 0.1.59 to 0.1.61

[dependabot[bot]]

Bumps jaxlib from 0.1.59 to 0.1.61.

Changelog

Sourced from jaxlib's changelog.

jaxlib 0.1.61 (February 12 2021)

jaxlib 0.1.60 (Febuary 3 2021)

• Bug fixes:
  • Fixed a memory leak when converting CPU DeviceArrays to NumPy arrays. The memory leak was present in jaxlib releases 0.1.58 and 0.1.59.
  • bool, int8, and uint8 are now considered safe to cast to bfloat16 NumPy extension type.

jax 0.2.9 (January 26 2021)

• GitHub commits.
• New features:
  • Extend the {mod}jax.experimental.loops module with support for pytrees. Improved error checking and error messages.
  • Add {func}jax.experimental.enable_x64 and {func}jax.experimental.disable_x64. These are context managers which allow X64 mode to be temporarily enabled/disabled within a session.
• Breaking changes:
  • {func}jax.ops.segment_sum now drops segment IDs that are out of range rather than wrapping them into the segment ID space. This was done for performance reasons.

Commits

• 8c3371c Merge pull request #5708 from skye:workspace
• 7eb6494 Merge pull request #5709 from skye:bazelversion
• 74db363 Bump required bazel version to 3.7.2
• 95444d9 Change TF version for jaxlib-v0.1.61
• b757792 Merge pull request #5703 from google:remove-extraneous-table-entry
• d135626 Merge pull request #5704 from google:make-jaxpr-shaped-inputs
• afecab9 accept any arguments with shape/dtype attributes after make_jaxpr
• 5357c6e remove float0 dtype from core.pytype_aval_mappings
• 4e6aff3 Merge pull request #5701 from jakevdp:expand-precommit
• 144a633 Merge pull request #5405 from shoyer:check-grads-err-msg
• 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)