Bump jaxlib from 0.1.56 to 0.1.62

[dependabot]

Bumps jaxlib from 0.1.56 to 0.1.62.

Changelog

Sourced from jaxlib's changelog.

jaxlib 0.1.62 (March 9 2021)

• New features:
  • jaxlib wheels are now built to require AVX instructions on x86-64 machines by default. If you want to use JAX on a machine that doesn't support AVX, you can build a jaxlib from source using the --target_cpu_features flag to build.py. --target_cpu_features also replaces --enable_march_native.

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.

jaxlib 0.1.59 (January 15 2021)

jax 0.2.8 (January 12 2021)

• GitHub commits.
• New features:
  • Add {func}jax.closure_convert for use with higher-order custom derivative functions. ({jax-issue}[#5244](https://github.com/google/jax/issues/5244))
  • Add {func}jax.experimental.host_callback.call to call a custom Python function on the host and return a result to the device computation. ({jax-issue}[#5243](https://github.com/google/jax/issues/5243))
• Bug fixes:
  • jax.numpy.arccosh now returns the same branch as numpy.arccosh for complex inputs ({jax-issue}[#5156](https://github.com/google/jax/issues/5156))
  • host_callback.id_tap now works for jax.pmap also. There is a optional parameter for id_tap and id_print to request that the device from which the value is tapped be passed as a keyword argument

... (truncated)

Commits

• a61c43c Merge pull request #5983 from skye:enable_tpu
• 591a484 Merge pull request #5908 from jakevdp:transforms-doc
• 749ad95 DOC: add transformations doc to HTML & reorganize contents
• 7b7a2a1 Don't build with Cloud TPU support for GPU or Mac wheels.
• 902038a Revert breaking change:
• 35214a1 Merge pull request #5962 from skye:cloud_tpu_env_vars
• 93d4779 Merge pull request #5980 from jakevdp:jax-101
• 3593323 Initial commit of seventh JAX-101 notebook
• a68e8b3 Merge pull request #5963 from jakevdp:jax-101-part6
• 0c139de Merge pull request #5964 from jakevdp:jax-101-authors
• 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 #184.