Add multi-host support to jax.experimental.sharded_jit
Add support for differentiating eigenvaleus computed by jax.numpy.linalg.eig
Add support for building on Windows platforms
Add support for general in_axes and out_axes in jax.pmap
Add complex support for jax.numpy.linalg.slogdet
Bug fixes:
Fix higher-than-second order derivatives of jax.numpy.sinc at zero
Fix some hard-to-hit bugs around symbolic zeros in transpose rules
Breaking changes:
jax.experimental.optix has been deleted, in favor of the standalone
optax Python package.
indexing of JAX arrays with non-tuple sequences now raises a TypeError. This type of indexing
has been deprecated in Numpy since v1.16, and in JAX since v0.2.4.
See [#4564](https://github.com/google/jax/issues/4564) <https://github.com/google/jax/pull/4564>_.
Commits
27a4c88 Merge pull request #5112 from google:update-pypi
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 jax from 0.2.6 to 0.2.7.
Changelog
Sourced from jax's changelog.
Commits
27a4c88
Merge pull request #5112 from google:update-pypi4664e82
update version and changelog for pypi7084e3d
Merge pull request #5111 from google:sda-repr8b64c3c
fix inherited repr method for ShardedDeviceArrayf776cdc
Merge pull request #5105 from google:device-put-replicated2f538cd9
Merge pull request #5095 from google:convert-element-type-old-dtype13b96cc
fix typo475d040
fix docstring tests2a8f71e
deflake tests6992ae1
switch assertions per reviewer commentDependabot 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)