{func}jax.lax.eigh now accepts an optional sort_eigenvalues argument
that allows users to opt out of eigenvalue sorting on TPU.
Deprecations
Non-array arguments to functions in {mod}jax.lax.linalg are now marked
keyword-only. As a backward-compatibility step passing keyword-only
arguments positionally yields a warning, but in a future JAX release passing
keyword-only arguments positionally will fail.
However, most users should prefer to use {mod}jax.numpy.linalg instead.
{func}jax.scipy.linalg.polar_unitary, which was a JAX extension to the
scipy API, is deprecated. Use {func}jax.scipy.linalg.polar instead.
{func}jax.mask {func}jax.shapecheck APIs have been removed.
See {jax-issue}[#11557](https://github.com/google/jax/issues/11557).
{mod}jax.experimental.loops has been removed. See {jax-issue}[#10278](https://github.com/google/jax/issues/10278)
for an alternative API.
{func}jax.tree_util.tree_multimap has been removed. It has been deprecated since
JAX release 0.3.5, and {func}jax.tree_util.tree_map is a direct replacement.
Removed jax.experimental.stax; it has long been a deprecated alias of
{mod}jax.example_libraries.stax.
Removed jax.experimental.optimizers; it has long been a deprecated alias of
{mod}jax.example_libraries.optimizers.
{func}jax.checkpoint, also known as {func}jax.remat, has a new
implementation switched on by default, meaning the old implementation is
deprecated; see JEP 11830.
JaxTestCase and JaxTestLoader have been removed from jax.test_util. These
classes have been deprecated since v0.3.1 ({jax-issue}[#11248](https://github.com/google/jax/issues/11248)).
Binary operations between JAX arrays and built-in collections (dict, list, set, tuple)
now raise a TypeError in all cases. Previously some cases (particularly equality and inequality)
would return boolean scalars inconsistent with similar operations in NumPy ({jax-issue}[#11234](https://github.com/google/jax/issues/11234)).
Several {mod}jax.tree_util routines accessed as top-level JAX package imports are now
deprecated, and will be removed in a future JAX release in accordance with the
{ref}api-compatibility policy:
{func}jax.treedef_is_leaf is deprecated in favor of {func}jax.tree_util.treedef_is_leaf
{func}jax.tree_flatten is deprecated in favor of {func}jax.tree_util.tree_flatten
{func}jax.tree_leaves is deprecated in favor of {func}jax.tree_util.tree_leaves
{func}jax.tree_structure is deprecated in favor of {func}jax.tree_util.tree_structure
{func}jax.tree_transpose is deprecated in favor of {func}jax.tree_util.tree_transpose
{func}jax.tree_unflatten is deprecated in favor of {func}jax.tree_util.tree_unflatten
The sym_pos argument of {func}jax.scipy.linalg.solve is deprecated in favor of assume_a='pos',
following a similar deprecation in {func}scipy.linalg.solve.
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
Bumps jax from 0.2.8 to 0.3.16.
Release notes
Sourced from jax's releases.
... (truncated)
Changelog
Sourced from jax's changelog.
... (truncated)
Commits
e9e074fMerge pull request #11862 from sharadmv:releaseb0309dcBump libtpu nightly version3d2026aCreate aSameDeviceAssignmentTupletype to cache the op shardings and devic...af18235Merge pull request #11856 from jakevdp:bug-templatef00dfb4issue template: avoid checkboxes because they're interpreted as tasksea67d9fMerge pull request #11852 from pschuh:unconstrained310c7a2Make vmap axis actually unconstrained by default.aaf1609Merge pull request #11850 from hawkinsp:ireedf74054Merge pull request #11846 from russbates:patch-1c9b03dfFix bitrot in IREE Python client.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)