Skip to content

Update jax requirement from <0.2.21,>0.2.16 to >0.2.16,<0.2.22

Vicentini Filippo requested to merge dependabot/pip/jax-gt-0.2.16-and-lt-0.2.22 into master

Created by: dependabot[bot]

Updates the requirements on jax to permit the latest version.

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.
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)

Merge request reports