Bump jaxlib from 0.1.55 to 0.1.56

[dependabot[bot]]

Bumps jaxlib from 0.1.55 to 0.1.56.

Changelog

Sourced from jaxlib's changelog.

Change Log

.. This is a comment. Remember to leave an empty line before the start of an itemized list, and to align the itemized text with the first line of an item.

.. PLEASE REMEMBER TO CHANGE THE '..master' WITH AN ACTUAL TAG in GITHUB LINK.

These are the release notes for JAX.

jax 0.2.2 (October 13 2020)

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

jax 0.2.1 (October 6 2020)

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

• Improvements:

  • As a benefit of omnistaging, the host_callback functions are executed (in program order) even if the result of the :py:func:jax.experimental.host_callback.id_print/ :py:func:jax.experimental.host_callback.id_tap is not used in the computation.

jax (0.2.0) (September 23 2020)

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

• Improvements:

  • Omnistaging on by default. See [#3370](https://github.com/google/jax/issues/3370) <https://github.com/google/jax/pull/3370>_.

jax (0.1.77) (September 15 2020)

• Breaking changes:

  • New simplified interface for :py:func:jax.experimental.host_callback.id_tap (#4101)

Commits

• 639cda2 Merge pull request #4554 from skye:workspace
• b8cac03 Update XLA in WORKSPACE
• 7c4b935 Merge pull request #4550 from jblespiau:changelist/336694640
• ee9ca56 Copybara import of the project:
• 5a097f5 Gate some jax_jit test with a version check.
• c1e2595 Add support for jax_debug_nans and fix the last few glitches with the C++ jax...
• 8f64715 Merge pull request #4546 from hawkinsp:flakes
• e1adbcd Fix batching_test flakiness on GPU.
• 9d8139e Merge pull request #4440 from google:long-line
• 7320fc7 Merge pull request #4542 from google:fix-jax2tf-typo
• 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)