search

asmith26 / jax_toolkit

A collection of jax functions to help with common machine/deep learning related functionality.
https://asmith26.github.io/jax_toolkit/
Apache License 2.0
5 stars 0 forks source link

Bump jax from 0.2.8 to 0.3.20 #342

Closed dependabot[bot] closed 1 year ago

dependabot[bot] commented 1 year ago

Bumps jax from 0.2.8 to 0.3.20.

Release notes

Sourced from jax's releases.

JAX release v0.3.19

Fixes the required jaxlib version

JAX release v0.3.18

  • GitHub commits.
  • Changes
    • Ahead-of-time lowering and compilation functionality (tracked in {jax-issue}[#7733](https://github.com/google/jax/issues/7733)) is stable and public. See the overview and the API docs for {mod}jax.stages.
    • Introduced {class}jax.Array, intended to be used for both isinstance checks and type annotations for array types in JAX. Notice that this included some subtle changes to how isinstance works for {class}jax.numpy.ndarray for jax-internal objects, as {class}jax.numpy.ndarray is now a simple alias of {class}jax.Array.
  • Breaking changes
    • jax._src is no longer imported into the from the public jax namespace. This may break users that were using JAX internals.
    • jax.soft_pmap has been deleted. Please use pjit or xmap instead. jax.soft_pmap is undocumented. If it were documented, a deprecation period would have been provided.

JAX release v0.3.17

  • GitHub commits.
  • Bugs
    • Fix corner case issue in gradient of lax.pow with an exponent of zero (#12041)
  • Breaking changes
    • jax.checkpoint, also known as jax.remat, no longer supports the concrete option, following the previous version's deprecation; see JEP 11830.
  • Changes
    • Added jax.pure_callback that enables calling back to pure Python functions from compiled functions (e.g. functions decorated with jax.jit or jax.pmap).
  • Deprecations:
    • The deprecated DeviceArray.tile() method has been removed. Use jax.numpy.tile (#11944).
    • DeviceArray.to_py() has been deprecated. Use np.asarray(x) instead.

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.

... (truncated)

Changelog

Sourced from jax's changelog.

jax 0.3.20

  • Bug fixes:
    • Adds missing .pyi files that were missing from the previous release ({jax-issue}[#12536](https://github.com/google/jax/issues/12536)).
    • Fixes an incompatibility between jax 0.3.19 and the libtpu version it pinned ({jax-issue}[#12550](https://github.com/google/jax/issues/12550)). Requires jaxlib 0.3.20.
    • Fix incorrect pip url in setup.py comment ({jax-issue}[#12528](https://github.com/google/jax/issues/12528)).

jaxlib 0.3.20

  • Bug fixes
    • Fixes support for limiting the visible CUDA devices via jax_cuda_visible_devices in distributed jobs. This functionality is needed for the JAX/SLURM integration on GPU ({jax-issue}[#12533](https://github.com/google/jax/issues/12533)).

jax 0.3.19 (Sep 27, 2022)

  • Fixes required jaxlib version.

jax 0.3.18 (Sep 26, 2022)

  • GitHub commits.
  • Changes
    • Ahead-of-time lowering and compilation functionality (tracked in {jax-issue}[#7733](https://github.com/google/jax/issues/7733)) is stable and public. See the overview and the API docs for {mod}jax.stages.
    • Introduced {class}jax.Array, intended to be used for both isinstance checks and type annotations for array types in JAX. Notice that this included some subtle changes to how isinstance works for {class}jax.numpy.ndarray for jax-internal objects, as {class}jax.numpy.ndarray is now a simple alias of {class}jax.Array.
  • Breaking changes
    • jax._src is no longer imported into the from the public jax namespace. This may break users that were using JAX internals.
    • jax.soft_pmap has been deleted. Please use pjit or xmap instead. jax.soft_pmap is undocumented. If it were documented, a deprecation period would have been provided.

jax 0.3.17 (Aug 31, 2022)

  • GitHub commits.
  • Bugs
    • Fix corner case issue in gradient of lax.pow with an exponent of zero ({jax-issue}12041)
  • Breaking changes
    • {func}jax.checkpoint, also known as {func}jax.remat, no longer supports the concrete option, following the previous version's deprecation; see JEP 11830.
  • Changes
    • Added {func}jax.pure_callback that enables calling back to pure Python functions from compiled functions (e.g. functions decorated with jax.jit or jax.pmap).
  • Deprecations:
    • The deprecated DeviceArray.tile() method has been removed. Use {func}jax.numpy.tile ({jax-issue}[#11944](https://github.com/google/jax/issues/11944)).
    • DeviceArray.to_py() has been deprecated. Use np.asarray(x) instead.

jax 0.3.16

... (truncated)

Commits
  • 0282b4b Merge pull request #12538 from jakevdp:bundle-pyi
  • aafc70d Merge pull request #12556 from hawkinsp:rocm
  • 5fe7a54 Merge pull request #12555 from hawkinsp:release
  • 39eabe8 Merge pull request #12552 from hawkinsp:nccl
  • f7bafb3 Disable multiprocess_gpu_test that fails on ROCm.
  • 8d86436 jax/jaxlib 0.3.20 release candidate.
  • eabb91e Fix test failure in GPU CI if NCCL_DEBUG is enabled.
  • 96abd9a Merge pull request #12540 from sharadmv:cond-lowering-fix
  • 96a85bd Make addressable_shards a property like local_shards
  • 9489068 Merge pull request #12546 from mattjj:issue12542
  • Additional commits viewable in compare view


Dependabot compatibility score

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 1 year ago

Superseded by #345.