Skip to content

Update jax requirement from <0.2.15,>=0.2.9 to >=0.2.9,<0.2.17

Vicentini Filippo requested to merge dependabot/pip/jax-gte-0.2.9-and-lt-0.2.17 into master

Created by: dependabot[bot]

Updates the requirements on jax to permit the latest version.

Changelog

Sourced from jax's changelog.

jax 0.2.16 (June 23 2021)

jax 0.2.15 (June 23 2021)

  • GitHub commits.

  • New features:

    • The {func}jax2tf.convert supports inequalities and min/max for booleans ({jax-issue}[#6956](https://github.com/google/jax/issues/6956)).

  • Breaking changes:

  • Bug fixes:

    • Fixed bug that prevented round-tripping from JAX to TF and back: jax2tf.call_tf(jax2tf.convert) ({jax-issue}[#6947](https://github.com/google/jax/issues/6947)).

jax 0.2.14 (June 10 2021)

  • GitHub commits.

  • New features:

    • The {func}jax2tf.convert now has support for pjit and sharded_jit.
    • A new configuration option JAX_TRACEBACK_FILTERING controls how JAX filters tracebacks.
    • A new traceback filtering mode using __tracebackhide__ is now enabled by default in sufficiently recent versions of IPython.
    • The {func}jax2tf.convert supports shape polymorphism even when the unknown dimensions are used in arithmetic operations, e.g., jnp.reshape(-1) ({jax-issue}[#6827](https://github.com/google/jax/issues/6827)).
    • The {func}jax2tf.convert generates custom attributes with location information in TF ops. The code that XLA generates after jax2tf has the same location information as JAX/XLA.

  • Bug fixes:

    • The {func}jax2tf.convert now ensures that it uses the same typing rules for Python scalars and for choosing 32-bit vs. 64-bit computations as JAX ({jax-issue}[#6883](https://github.com/google/jax/issues/6883)).
    • The {func}jax2tf.convert now scopes the enable_xla conversion parameter properly to apply only during the just-in-time conversion ({jax-issue}[#6720](https://github.com/google/jax/issues/6720)).
    • The {func}jax2tf.convert now converts lax.dot_general using the XlaDot TensorFlow op, for better fidelity w.r.t. JAX numerical precision ({jax-issue}[#6717](https://github.com/google/jax/issues/6717)).
    • The {func}jax2tf.convert now has support for inequality comparisons and min/max for complex numbers ({jax-issue}[#6892](https://github.com/google/jax/issues/6892)).

jaxlib 0.1.67 (unreleased)

jaxlib 0.1.66 (May 11 2021)

  • New features:

... (truncated)

Commits
  • c509731 Merge pull request #7084 from skye:version
  • 3da8a4c Update jax version to 0.2.16
  • c985d76 Merge pull request #7082 from zhangqiaorjc:gtsv2fix
  • 4c1856e Merge pull request #7083 from skye:fix_install
  • 7cc277d Test for gtsv2 attr on cusparse.
  • 55276d1 Fix pip install jax[tpu]
  • 2460f91 Revert "Improve support for pip install jax[cuda111]"
  • 9493b31 Merge pull request #7080 from jakevdp:test-dynamic-slice
  • 5bfcb51 Merge pull request #7059 from zhangqiaorjc:cpp_fix
  • 27fc797 Add dynamic slice U8 index test
  • 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