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.3.21 to 0.4.6 #143

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

dependabot[bot]
Copy link

@dependabot dependabot bot commented on behalf of github Mar 10, 2023

Bumps jax[cpu] from 0.3.21 to 0.4.6.

Release notes

Sourced from jax[cpu]'s releases.

JAX release v0.4.6

No release notes provided.

jaxlib release v0.4.6

No release notes provided.

JAX release v0.4.5

jax version 0.4.5

Jax release v0.4.4

No release notes provided.

Jaxlib release v0.4.4

No release notes provided.

JAX release v0.4.3

No release notes provided.

jaxlib release v0.4.3

No release notes provided.

JAX release v0.4.2

No release notes provided.

jaxlib release v0.4.2

No release notes provided.

Jax release v0.4.1

  • Changes
    • Support for Python 3.7 has been dropped, in accordance with JAX's {ref}version-support-policy.
    • We introduce jax.Array which is a unified array type that subsumes DeviceArray, ShardedDeviceArray, and GlobalDeviceArray types in JAX. The jax.Array type helps make parallelism a core feature of JAX, simplifies and unifies JAX internals, and allows us to unify jit and pjit. jax.Array has been enabled by default in JAX 0.4 and makes some breaking change to the pjit API. The jax.Array migration guide can help you migrate your codebase to jax.Array. You can also look at the Distributed arrays and automatic parallelization tutorial to understand the new concepts.
    • PartitionSpec and Mesh are now out of experimental. The new API endpoints are jax.sharding.PartitionSpec and jax.sharding.Mesh. jax.experimental.maps.Mesh and jax.experimental.PartitionSpec are deprecated and will be removed in 3 months.
    • with_sharding_constraints new public endpoint is jax.lax.with_sharding_constraint.
    • If using ABSL flags together with jax.config, the ABSL flag values are no longer read or written after the JAX configuration options are initially populated from the ABSL flags. This change improves performance of reading

... (truncated)

Changelog

Sourced from jax[cpu]'s changelog.

jax 0.4.6 (Mar 9, 2023)

  • Changes

    • jax.tree_util now contain a set of APIs that allow user to define keys for their custom pytree node. This includes:

      • tree_flatten_with_path that flattens a tree and return not only each leaf but also their key paths.
      • tree_map_with_paths that can map a function that takes the key path as argument.
      • `register_pytree_with_keys`` to register how the key path and leaves should looks like in a custom pytree node.
      • keystr that pretty-prints a key path.
    • {func}jax2tf.call_tf has a new parameter output_shape_dtype (default None) that can be used to declare the output shape and type of the result. This enables {func}jax2tf.call_tf to work in the presence of shape polymorphism. ({jax-issue}[#14734](https://github.com/google/jax/issues/14734)).

  • Deprecations

    • The old key-path APIs in jax.tree_util are deprecated and will be removed 3 months from Mar 10 2023:
      • register_keypaths: use {func}jax.tree_util.register_pytree_with_keys instead.
      • AttributeKeyPathEntry : use GetAttrKey instead.
      • GetitemKeyPathEntry : use SequenceKey or DictKey instead.

jaxlib 0.4.6 (Mar 9, 2023)

jax 0.4.5 (Mar 2, 2023)

  • Deprecations
    • jax.sharding.OpShardingSharding has been renamed to jax.sharding.GSPMDSharding. jax.sharding.OpShardingSharding will be removed in 3 months from Feb 17, 2023.
    • The following jax.Array methods are deprecated and will be removed 3 months from Feb 23 2023:
      • jax.Array.broadcast: use {func}jax.lax.broadcast instead.
      • jax.Array.broadcast_in_dim: use {func}jax.lax.broadcast_in_dim instead.
      • jax.Array.split: use {func}jax.numpy.split instead.

jax 0.4.4 (Feb 16, 2023)

  • Changes
    • The implementation of jit and pjit has been merged. Merging jit and pjit changes the internals of JAX without affecting the public API of JAX. Before, jit was a final style primitive. Final style means that the creation of jaxpr was delayed as much as possible and transformations were stacked on top of each other. With the jit-pjit implementation merge, jit becomes an initial style primitive which means that we trace to jaxpr as early as possible. For more information see this section in autodidax. Moving to initial style should simplify JAX's internals and make development of features like dynamic shapes, etc easier. You can disable it only via the environment variable i.e.

... (truncated)

Commits
  • f1f4840 Merge pull request #14881 from skye:version
  • 1aa08fd Update WORKSPACE and setup.py for jax/jaxlib 0.4.6 release
  • 7fd1e2f Split _src/traceback_util.py into its own Bazel target.
  • 01b00c4 Increase sharding of shard_map test on CPU.
  • 9912a8e Split _src/pretty_printer.py into its own Bazel target.
  • d216d98 Remove the disassemble into single devices arrays in ExecuteReplicated.__call...
  • 08789fd Exclude "util.py" and "config.py" from the main JAX bazel target.
  • 560fe73 [jax2tf] Disable some failing tests
  • 0e05a79 Split some submodules out of //jax under Bazel.
  • 5c91453 [jax2tf] Add check that native lowering should not include custom calls not g...
  • 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)

Bumps [jax[cpu]](https://github.com/google/jax) from 0.3.21 to 0.4.6.
- [Release notes](https://github.com/google/jax/releases)
- [Changelog](https://github.com/google/jax/blob/main/CHANGELOG.md)
- [Commits](jax-ml/jax@0.3.21...jax-v0.4.6)

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

Signed-off-by: dependabot[bot] <support@github.com>
@dependabot dependabot bot added dependencies Pull requests that update a dependency file python Pull requests that update Python code labels Mar 10, 2023
@codecov-commenter
Copy link

codecov-commenter commented Mar 10, 2023

Codecov Report

Merging #143 (dc289f0) into main (20b01a3) will decrease coverage by 17.90%.
The diff coverage is n/a.

📣 This organization is not using Codecov’s GitHub App Integration. We recommend you install it so Codecov can continue to function properly for your repositories. Learn more

Additional details and impacted files

Impacted file tree graph

@@             Coverage Diff             @@
##             main     #143       +/-   ##
===========================================
- Coverage   85.90%   68.01%   -17.90%     
===========================================
  Files         248      275       +27     
  Lines       23312    24158      +846     
  Branches     4212     4322      +110     
===========================================
- Hits        20027    16431     -3596     
- Misses       2224     6851     +4627     
+ Partials     1061      876      -185     

see 104 files with indirect coverage changes

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.

1 participant