Bump jax from 0.2.5 to 0.2.21

[dependabot]

Bumps jax from 0.2.5 to 0.2.21.

Release notes

Sourced from jax's releases.

Jax release v0.2.21

• New features:

  • Added jax.numpy.insert implementation (#7936 ).

• Breaking Changes

  • jax.api has been removed. Functions that were available as jax.api.* were aliases for functions in jax.*; please use the functions in jax.* instead.
  • jax.partial, jax.lax.partial, and jax.util.partial were accidental exports that have now been removed. Use functools.partial from the Python standard library instead.
  • Boolean scalar indices now raise a TypeError; previously this silently returned wrong results (#7925 ).
  • Many more jax.numpy functions now require array-like inputs, and will error if passed a list (#7747 #7802 #7907 ). See #7737 for a discussion of the rationale behind this change.
  • When inside a transformation such as jax.jit, jax.numpy.array always stages the array it produces into the traced computation. Previously jax.numpy.array would sometimes produce a on-device array, even under a jax.jit decorator. This change may break code that used JAX arrays to perform shape or index computations that must be known statically; the workaround is to perform such computations using classic NumPy arrays instead.

Jax release v0.2.20

No release notes provided.

Changelog

Sourced from jax's changelog.

jax 0.2.21 (Sept 23, 2021)

• GitHub commits.
• Breaking Changes
  • jax.api has been removed. Functions that were available as jax.api.* were aliases for functions in jax.*; please use the functions in jax.* instead.
  • jax.partial, jax.lax.partial, and jax.util.partial were accidental exports that have now been removed. Use functools.partial from the Python standard library instead.
  • Boolean scalar indices now raise a TypeError; previously this silently returned wrong results ({jax-issue}[#7925](https://github.com/google/jax/issues/7925)).
  • Many more jax.numpy functions now require array-like inputs, and will error if passed a list ({jax-issue}[#7747](https://github.com/google/jax/issues/7747) {jax-issue}[#7802](https://github.com/google/jax/issues/7802) {jax-issue}[#7907](https://github.com/google/jax/issues/7907)). See {jax-issue}[#7737](https://github.com/google/jax/issues/7737) for a discussion of the rationale behind this change.
  • When inside a transformation such as jax.jit, jax.numpy.array always stages the array it produces into the traced computation. Previously jax.numpy.array would sometimes produce a on-device array, even under a jax.jit decorator. This change may break code that used JAX arrays to perform shape or index computations that must be known statically; the workaround is to perform such computations using classic NumPy arrays instead.
• New features:
  • Added {func}jax.numpy.insert implementation ({jax-issue}[#7936](https://github.com/google/jax/issues/7936)).

jax 0.2.20 (Sept 2, 2021)

• GitHub commits.
• Breaking Changes
  • jnp.poly* functions now require array-like inputs ({jax-issue}[#7732](https://github.com/google/jax/issues/7732))
  • jnp.unique and other set-like operations now require array-like inputs ({jax-issue}[#7662](https://github.com/google/jax/issues/7662))

jaxlib 0.1.71 (Sep 1, 2021)

• Breaking changes:
  • Support for CUDA 11.0 and CUDA 10.1 has been dropped. Jaxlib now supports CUDA 10.2 and CUDA 11.1+.

jax 0.2.19 (Aug 12, 2021)

• GitHub commits.
• Breaking changes:

  • Support for NumPy 1.17 has been dropped, per the deprecation policy. Please upgrade to a supported NumPy version.

  • The jit decorator has been added around the implementation of a number of operators on JAX arrays. This speeds up dispatch times for common operators such as +.

    This change should largely be transparent to most users. However, there is

... (truncated)

Commits

• dbeb97d Create 0.2.21 jax release
• b6318f8 Merge pull request #7993 from gnecula:tf_share_const
• b992034 Merge pull request #7983 from jakevdp:fix-std-basis
• b1ad9e3 [jax2tf] Ensure that shared constants in JAX are shared also in the converted...
• 9389db1 Merge pull request #7975 from google:custom-derivatives-api-boundary
• 2c2f403 Move contents of jax.lib to jax._src.lib.
• 94cd1ea Merge pull request #7978 from gnecula:tf_comment
• 322a954 disable pytype on custom derivative calls
• 0957e81 Use traced identity in jacobian std_basis
• 484b5af Merge pull request #7982 from hawkinsp:checktol
• 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 #246.