Bump jax from 0.2.8 to 0.2.16

[dependabot[bot]]

Bumps jax from 0.2.8 to 0.2.16.

Changelog

Sourced from jax's changelog.

jax 0.2.16 (June 23 2021)

• GitHub commits.

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:

  • Support for NumPy 1.16 has been dropped, per the deprecation policy.

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

[dependabot[bot]]

Superseded by #210.