Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Bump jax[cpu] from 0.4.8 to 0.4.22 #2351

Closed
wants to merge 1 commit into from

Conversation

dependabot[bot]
Copy link
Contributor

@dependabot dependabot bot commented on behalf of github Dec 14, 2023

Bumps jax[cpu] from 0.4.8 to 0.4.22.

Release notes

Sourced from jax[cpu]'s releases.

Jax release v0.4.22

No release notes provided.

Jaxlib release v0.4.22

No release notes provided.

Jax release v0.4.21

No release notes provided.

Jaxlib release v0.4.21

No release notes provided.

JAX release v0.4.20

No release notes provided.

jaxlib release v0.4.20

No release notes provided.

Jax release v0.4.19

No release notes provided.

jaxlib release v0.4.19

No release notes provided.

JAX release v0.4.18

No release notes provided.

jaxlib-v0.4.18

No release notes provided.

JAX release v0.4.17

No release notes provided.

jaxlib release v0.4.17

No release notes provided.

Jax release v0.4.16

No release notes provided.

Jaxlib release v0.4.16

No release notes provided.

JAX release v0.4.14

No release notes provided.

jaxlib release v0.4.14

No release notes provided.

JAX release v0.4.13

NOTE: This is the last JAX release that will include Python 3.8 support

... (truncated)

Changelog

Sourced from jax[cpu]'s changelog.

jax 0.4.22 (Dec 13, 2023)

  • Deprecations
    • The device_buffer and device_buffers properties of JAX arrays are deprecated. Explicit buffers have been replaced by the more flexible array sharding interface, but the previous outputs can be recovered this way:
      • arr.device_buffer becomes arr.addressable_data(0)
      • arr.device_buffers becomes [x.data for x in arr.addressable_shards]

jaxlib 0.4.22 (Dec 13, 2023)

jax 0.4.21 (Dec 4 2023)

  • New Features

    • Added {obj}jax.nn.squareplus.
  • Changes

    • The minimum jaxlib version is now 0.4.19.
    • Released wheels are built now with clang instead of gcc.
    • Enforce that the device backend has not been initialized prior to calling jax.distributed.initialize().
    • Automate arguments to jax.distributed.initialize() in cloud TPU environments.
  • Deprecations

    • The previously-deprecated sym_pos argument has been removed from {func}jax.scipy.linalg.solve. Use assume_a='pos' instead.
    • Passing None to {func}jax.array or {func}jax.asarray, either directly or within a list or tuple, is deprecated and now raises a {obj}FutureWarning. It currently is converted to NaN, and in the future will raise a {obj}TypeError.
    • Passing the condition, x, and y parameters to jax.numpy.where by keyword arguments has been deprecated, to match numpy.where.
    • Passing arguments to {func}jax.numpy.array_equal and {func}jax.numpy.array_equiv that cannot be converted to a JAX array is deprecated and now raises a {obj}DeprecationWaning. Currently the functions return False, in the future this will raise an exception.
    • The device() method of JAX arrays deprecated. Depending on the context, it may be replaced with one of the following:
      • {meth}jax.Array.devices returns the set of all devices used by the array.
      • {attr}jax.Array.sharding gives the sharding configuration used by the array.

jaxlib 0.4.21 (Dec 4 2023)

  • Changes
    • In preparation for adding distributed CPU support, JAX now treats CPU devices identically to GPU and TPU devices, that is:

      • jax.devices() includes all devices present in a distributed job, even those not local to the current process. jax.local_devices() still only includes devices local to the current process, so if the change to jax.devices() breaks you, you most likely want to use jax.local_devices() instead.

... (truncated)

Commits
  • b52bcc1 Reverts 3c07c10a9a55f9a32390dd10cf3f420bdf3f1ed8
  • e888806 Fix the pjit flakey test. The test was weirdly written in the first place. Th...
  • 2e63352 Start jax and jaxlib 0.4.22 release
  • 3c07c10 Remove the `jax_require_devices_during_lowering flag since it was temporary. ...
  • 0281596 Merge pull request #18869 from 8bitmp3:small-fix-jax-pytrees
  • d269c58 Merge pull request #18582 from 8bitmp3:jax-docs-debugging-101
  • 4459991 Merge pull request #18961 from mattjj:issue18955
  • 196c97f Merge pull request #18949 from froystig:seed-offset
  • 5104c6b Merge pull request #18951 from mattjj:shmap-varargs-error
  • 4ba6bd5 [shard-map] register cumsum et al with generic rules
  • 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 show <dependency name> ignore conditions will show all of the ignore conditions of the specified dependency
  • @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[cpu]](https://github.com/google/jax) from 0.4.8 to 0.4.22.
- [Release notes](https://github.com/google/jax/releases)
- [Changelog](https://github.com/google/jax/blob/main/CHANGELOG.md)
- [Commits](jax-ml/jax@jax-v0.4.8...jax-v0.4.22)

---
updated-dependencies:
- dependency-name: jax[cpu]
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <[email protected]>
@dependabot dependabot bot added dependencies Pull requests that update a dependency file python Pull requests that update Python code labels Dec 14, 2023
@codecov-commenter
Copy link

codecov-commenter commented Dec 14, 2023

Codecov Report

Merging #2351 (0eddcb5) into main (ffd3a49) will decrease coverage by 3.03%.
The diff coverage is n/a.

❗ Your organization needs to install the Codecov GitHub app to enable full functionality.

Additional details and impacted files

Impacted file tree graph

@@            Coverage Diff             @@
##             main    #2351      +/-   ##
==========================================
- Coverage   81.39%   78.36%   -3.03%     
==========================================
  Files         324      324              
  Lines       29331    29331              
  Branches     5409     5409              
==========================================
- Hits        23873    22986     -887     
- Misses       4092     5053     +961     
+ Partials     1366     1292      -74     

see 47 files with indirect coverage changes

Copy link
Contributor Author

dependabot bot commented on behalf of github Dec 14, 2023

Superseded by #2354.

@dependabot dependabot bot closed this Dec 14, 2023
@dependabot dependabot bot deleted the dependabot/pip/jax-cpu--0.4.22 branch December 14, 2023 12:04
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
dependencies Pull requests that update a dependency file python Pull requests that update Python code
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants