Bump jax from 0.2.8 to 0.2.17

[dependabot[bot]]

Bumps jax from 0.2.8 to 0.2.17.

Changelog

Sourced from jax's changelog.

jax 0.2.17 (July 9 2021)

• GitHub commits.
• Bug fixes:
  • Default to the older "stream_executor" CPU runtime for jaxlib <= 0.1.68 to work around #7229, which caused wrong outputs on CPU due to a concurrency problem.
• New features:
  • New SciPy function {py:func}jax.scipy.special.sph_harm.
  • Reverse-mode autodiff functions ({func}jax.grad, {func}jax.value_and_grad, {func}jax.vjp, and {func}jax.linear_transpose) support a parameter that indicates which named axes should be summed over in the backward pass if they were broadcasted over in the forward pass. This enables use of these APIs in a non-per-example way inside maps (initially only {func}jax.experimental.maps.xmap) ({jax-issue}[#6950](https://github.com/google/jax/issues/6950)).

jax 0.2.16 (June 23 2021)

• GitHub commits.

jax 0.2.15 (June 23 2021)

• GitHub commits.

• New features:

  • #7042 Turned on TFRT CPU backend with significant dispatch performance improvements on CPU.
  • The {func}jax2tf.convert supports inequalities and min/max for booleans ({jax-issue}[#6956](https://github.com/google/jax/issues/6956)).
  • New SciPy function {py:func}jax.scipy.special.lpmn_values.

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

jaxlib 0.1.68 (June 23 2021)

• Bug fixes:
  • Fixed bug in TFRT CPU backend that gets nans when transfer TPU buffer to CPU.

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.

... (truncated)

Commits

• 83368aa Merge pull request #7230 from hawkinsp:jaxrelease
• b393d9a Update jax version and changelog for 0.1.27.
• a50dd08 Merge pull request #7226 from jakevdp:sparse-rdot
• fd7b83f Merge pull request #7221 from jakevdp:vander-rank-promotion
• 591c315 Fix rank promotion warning in jnp.vander
• d2861b1 Merge pull request #7114 from charmerDark:s_implementation
• 087da55 [sparse]: add support for rdot_general in sparsify transform
• 6cc32ef Merge pull request #7224 from jakevdp:fix-ci
• b178bd1 Fix testExpandDims for older numpy
• cb51acc Merge pull request #7218 from jakevdp:expand-dims
• 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 #213.