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).
jaxlib 0.3.15 (Unreleased)
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)
Bumps jax from 0.3.14 to 0.3.15.
Changelog
Sourced from jax's changelog.
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)