Bump jaxlib from 0.1.61 to 0.1.66

[dependabot[bot]]

Bumps jaxlib from 0.1.61 to 0.1.66.

Changelog

Sourced from jaxlib's changelog.

jaxlib 0.1.66 (May 11 2021)

• New features:

  • CUDA 11.1 wheels are now supported on all CUDA 11 versions 11.1 or higher.

    NVidia now promises compatibility between CUDA minor releases starting with CUDA 11.1. This means that JAX can release a single CUDA 11.1 wheel that is compatible with CUDA 11.2 and 11.3.

    There is no longer a separate jaxlib release for CUDA 11.2 (or higher); use the CUDA 11.1 wheel for those versions (cuda111).

  • Added support for static keyword arguments to the C++ jit implementation.

jax 0.2.13 (May 3 2021)

• GitHub commits.
• New features:
  • When combined with jaxlib 0.1.66, {func}jax.jit now supports static keyword arguments. A new static_argnames option has been added to specify keyword arguments as static.
  • {func}jax.nonzero has a new optional size argument that allows it to be used within jit ({jax-issue}[#6501](https://github.com/google/jax/issues/6501))
  • {func}jax.numpy.unique now supports the axis argument ({jax-issue}[#6532](https://github.com/google/jax/issues/6532)).
  • {func}jax.experimental.host_callback.call now supports pjit.pjit ({jax-issue}[#6569](https://github.com/google/jax/issues/6569)).
  • Added {func}jax.scipy.linalg.eigh_tridiagonal that computes the eigenvalues of a tridiagonal matrix. Only eigenvalues are supported at present.
• Breaking changes:
  • The following function names have changed. There are still aliases, so this should not break existing code, but the aliases will eventually be removed so please change your code.
    • host_id --> {func}~jax.process_index
    • host_count --> {func}~jax.process_count
    • host_ids --> range(jax.process_count())
  • Similarly, the argument to {func}~jax.local_devices has been renamed from host_id to process_index.
  • Arguments to {func}jax.jit other than the function are now marked as keyword-only. This change is to prevent accidental breakage when arguments are added to jit.
• Bug fixes:
  • The {func}jax2tf.convert now works in presence of gradients for functions with integer inputs ({jax-issue}[#6360](https://github.com/google/jax/issues/6360)).
  • Fixed assertion failure in {func}jax2tf.call_tf when used with captured tf.Variable ({jax-issue}[#6572](https://github.com/google/jax/issues/6572)).

jaxlib 0.1.65 (April 7 2021)

jax 0.2.12 (April 1 2021)

• GitHub commits.
• New features
  • New profiling APIs: {func}jax.profiler.start_trace, {func}jax.profiler.stop_trace, and {func}jax.profiler.trace

... (truncated)

Commits

• f7717c8 Merge pull request #6724 from hawkinsp:dispatch
• e42fc2e Merge pull request #6712 from hawkinsp:fb
• 0f4720b Merge pull request #6543 from cdfreeman-google:master
• d77be6e Merge pull request #6716 from NeilGirdhar:annotations
• aea068e Reenable numpy dispatch tests.
• 4f9eb64 [XLA] Fix incomplete gamma functions where x is infinity.
• d724a30 Use Array instead of jnp.array in jax.random
• 078d8e1 Merge pull request #6706 from hawkinsp:indexops
• 01d6e32 Add version constraints to flatbuffers versions.
• d005e38 Promote the x.at[idx].set(y) operators as the preferred way to do indexed upd...
• 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[bot]]

Superseded by #194.