Skip to content

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

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

Created by: dependabot[bot]

Updates the requirements on jax to permit the latest version.

Release notes

Sourced from jax's releases.

JAX release v0.3.16

  • GitHub commits.
  • Breaking changes
    • Support for NumPy 1.19 has been dropped, per the deprecation policy. Please upgrade to NumPy 1.20 or newer.
  • Changes
    • Added jax.debug that includes utilities for runtime value debugging such at jax.debug.print and jax.debug.breakpoint.
    • Added new documentation for runtime value debugging
  • Deprecations
    • jax.mask jax.shapecheck APIs have been removed. See #11557.
    • jax.experimental.loops has been removed. See #10278 for an alternative API.
    • jax.tree_util.tree_multimap has been removed. It has been deprecated since JAX release 0.3.5, and jax.tree_util.tree_map is a direct replacement.
    • Removed jax.experimental.stax; it has long been a deprecated alias of jax.example_libraries.stax.
    • Removed jax.experimental.optimizers; it has long been a deprecated alias of jax.example_libraries.optimizers.
    • jax.checkpoint, also known as jax.remat, has a new implementation switched on by default, meaning the old implementation is deprecated; see JEP 11830.
Changelog

Sourced from jax's changelog.

jax 0.3.16 (Unreleased)

  • GitHub commits.
  • Breaking changes
    • Support for NumPy 1.19 has been dropped, per the deprecation policy. Please upgrade to NumPy 1.20 or newer.
  • Changes
    • Added {mod}jax.debug that includes utilities for runtime value debugging such at {func}jax.debug.print and {func}jax.debug.breakpoint.
    • Added new documentation for runtime value debugging
  • Deprecations
    • {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.

jaxlib 0.3.16 (Unreleased)

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)

... (truncated)

Commits
  • e9e074f Merge pull request #11862 from sharadmv:release
  • b0309dc Bump libtpu nightly version
  • 3d2026a Create a SameDeviceAssignmentTuple type to cache the op shardings and devic...
  • af18235 Merge pull request #11856 from jakevdp:bug-template
  • f00dfb4 issue template: avoid checkboxes because they're interpreted as tasks
  • ea67d9f Merge pull request #11852 from pschuh:unconstrained
  • 310c7a2 Make vmap axis actually unconstrained by default.
  • aaf1609 Merge pull request #11850 from hawkinsp:iree
  • df74054 Merge pull request #11846 from russbates:patch-1
  • c9b03df Fix bitrot in IREE Python client.
  • 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