{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.
Jax release v0.3.10
No release notes provided.
Jaxlib release v0.3.10
No release notes provided.
Jax release 0.3.9
Changes
Added support for fully asynchronous checkpointing for GlobalDeviceArray.
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)).
{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_map is deprecated in favor of {func}jax.tree_util.tree_map
{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.
Deprecations:
{func}jax.mask {func}jax.shapecheck APIs have been removed.
See {jax-issue}[#11557](https://github.com/google/jax/issues/11557).
{func}jax.experimental.compilation_cache.initialize_cache does not support
max_cache_size_ bytes anymore and will not get that as an input.
JAX_PLATFORMS now raises an exception when platform initialization fails.
Changes
Fixed compatibility problems with NumPy 1.23.
{func}jax.numpy.linalg.slogdet now accepts an optional method argument
that allows selection between an LU-decomposition based implementation and
an implementation based on QR decomposition.
{func}jax.numpy.linalg.qr now supports mode="raw".
pickle, copy.copy, and copy.deepcopy now have more complete support when
used on jax arrays ({jax-issue}[#10659](https://github.com/google/jax/issues/10659)). In particular:
pickle and deepcopy previously returned np.ndarray objects when used
on a DeviceArray; now DeviceArray objects are returned. For deepcopy,
the copied array is on the same device as the original. For pickle the
deserialized array will be on the default device.
Within function transformations (i.e. traced code), deepcopy and copy
previously were no-ops. Now they use the same mechanism as DeviceArray.copy().
Calling pickle on a traced array now results in an explicit
ConcretizationTypeError.
The implementation of singular value decomposition (SVD) and
symmetric/Hermitian eigendecomposition should be significantly faster on
... (truncated)
Commits
1a7c883 Merge pull request #11589 from skye:workspace
26fbeb6 Update WORKSPACE and libtpu version for jaxlib 0.3.15, take 3
e121e81 Merge pull request #11536 from sharadmv:colab-debugger
0b6657e Merge pull request #11556 from RuffaloLavoisier:tYpO
4870710 Enable debugging callbacks with pjit on TPU
8a67734 Merge pull request #11579 from sharadmv:fix-effects
7f0b917 Merge pull request #11575 from gnecula:ds_progress
24134ec Merge pull request #11425 from pschuh:pjit-bugfix
540ee56 Merge pull request #11576 from jakevdp:searchsorted-alt
d6c172d Fix PE not allowing double JIT-ted effectful functions
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.15.
Release notes
Sourced from jax's releases.
... (truncated)
Changelog
Sourced from jax's changelog.
... (truncated)
Commits
1a7c883Merge pull request #11589 from skye:workspace26fbeb6Update WORKSPACE and libtpu version for jaxlib 0.3.15, take 3e121e81Merge pull request #11536 from sharadmv:colab-debugger0b6657eMerge pull request #11556 from RuffaloLavoisier:tYpO4870710Enable debugging callbacks with pjit on TPU8a67734Merge pull request #11579 from sharadmv:fix-effects7f0b917Merge pull request #11575 from gnecula:ds_progress24134ecMerge pull request #11425 from pschuh:pjit-bugfix540ee56Merge pull request #11576 from jakevdp:searchsorted-altd6c172dFix PE not allowing double JIT-ted effectful functionsDependabot 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)