Ahead-of-time lowering and compilation functionality (tracked in
{jax-issue}[#7733](https://github.com/google/jax/issues/7733)) is stable and public. See the
overview and the API docs
for {mod}jax.stages.
Introduced {class}jax.Array, intended to be used for both isinstance checks
and type annotations for array types in JAX. Notice that this included some subtle
changes to how isinstance works for {class}jax.numpy.ndarray for jax-internal
objects, as {class}jax.numpy.ndarray is now a simple alias of {class}jax.Array.
Breaking changes
jax._src is no longer imported into the from the public jax namespace.
This may break users that were using JAX internals.
jax.soft_pmap has been deleted. Please use pjit or xmap instead.
jax.soft_pmap is undocumented. If it were documented, a deprecation period
would have been provided.
Fix corner case issue in gradient of lax.pow with an exponent of zero
({jax-issue}12041)
Breaking changes
{func}jax.checkpoint, also known as {func}jax.remat, no longer supports
the concrete option, following the previous version's deprecation; see
JEP 11830.
Changes
Added {func}jax.pure_callback that enables calling back to pure Python functions from compiled functions (e.g. functions decorated with jax.jit or jax.pmap).
Deprecations:
The deprecated DeviceArray.tile() method has been removed. Use {func}jax.numpy.tile
({jax-issue}[#11944](https://github.com/google/jax/issues/11944)).
DeviceArray.to_py() has been deprecated. Use np.asarray(x) instead.
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]
commented
2 years ago
Updates the requirements on jax to permit the latest version.
Release notes
Sourced from jax's releases.
Changelog
Sourced from jax's changelog.
... (truncated)
Commits
9e4114fMovearray.pyandsharding.pyfromexperimental/to_src/.0e11688Merge pull request #12382 from jakevdp:reduction-dtype1bcf8d6Merge pull request #12497 from mattjj:djax-dag-fix1e42247bMerge pull request #12524 from sharadmv:lax-import-fix389a2e5Add a backwards compat path forop_sharding.clone()because it doesn't exis...1e7ca8ffix bug in djax type signature inference logic1d895b2Fix lax importscbf34cbRename the concrete classArraytoArrayImpl71bcabe[sparse] Add BCSR format template.82636b0Merge pull request #12523 from jakevdp:fix-buildDependabot 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)