Skip to content

Update jax requirement from <0.3.15,>=0.3.2 to >=0.3.2,<0.3.16

Vicentini Filippo requested to merge dependabot/pip/jax-gte-0.3.2-and-lt-0.3.16 into master

Created by: dependabot[bot]

Updates the requirements on jax to permit the latest version.

Changelog

Sourced from jax's changelog.

jax 0.3.15 (July 22, 2022)

  • GitHub commits.
  • Changes
    • 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)).
    • Added {class}jax.scipy.gaussian_kde ({jax-issue}[#11237](https://github.com/google/jax/issues/11237)).
    • 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.

jaxlib 0.3.15 (July 22, 2022)

jax 0.3.14 (June 27, 2022)

  • GitHub commits.
  • Breaking changes
    • {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 TPU, especially for matrices above 1000x1000 or so. Both now use a spectral divide-and-conquer algorithm for eigendecomposition (QDWH-eig).

... (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
  • Additional commits viewable in compare view

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)

Merge request reports