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)).
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.
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)
Bumps jaxlib from 0.1.61 to 0.1.69.
Changelog
Sourced from jaxlib's changelog.
... (truncated)
Commits
fa758bc
Merge pull request #7233 from zhangqiaorjc:update_xla_2991aba6
Update XLA.83368aa
Merge pull request #7230 from hawkinsp:jaxreleaseb393d9a
Update jax version and changelog for 0.1.27.a50dd08
Merge pull request #7226 from jakevdp:sparse-rdotfd7b83f
Merge pull request #7221 from jakevdp:vander-rank-promotion591c315
Fix rank promotion warning in jnp.vanderd2861b1
Merge pull request #7114 from charmerDark:s_implementation087da55
[sparse]: add support for rdot_general in sparsify transform6cc32ef
Merge pull request #7224 from jakevdp:fix-ciDependabot 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)