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

Update jax requirement from <0.2.19,>0.2.16 to >0.2.16,<0.4.6 #72

Closed
wants to merge 1 commit into from

Conversation

dependabot[bot]
Copy link

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

Updates the requirements on jax to permit the latest version.

Changelog

Sourced from jax's changelog.

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. os.environ['JAX_JIT_PJIT_API_MERGE'] = '0'. The merge must be disabled via an environment variable since it affects JAX at import time so it needs to be disabled before jax is imported.
    • axis_resources argument of with_sharding_constraint is deprecated. Please use shardings instead. There is no change needed if you were using axis_resources as an arg. If you were using it as a kwarg, then please use shardings instead. axis_resources will be removed after 3 months from Feb 13, 2023.
    • added the {mod}jax.typing module, with tools for type annotations of JAX functions.
    • The following names have been deprecated:
      • jax.xla.Device and jax.interpreters.xla.Device: use jax.Device.
      • jax.experimental.maps.Mesh. Use jax.sharding.Mesh instead.
      • jax.experimental.pjit.NamedSharding: use jax.sharding.NamedSharding.
      • jax.experimental.pjit.PartitionSpec: use jax.sharding.PartitionSpec.
      • jax.interpreters.pxla.Mesh: use jax.sharding.Mesh.
      • jax.interpreters.pxla.PartitionSpec: use jax.sharding.PartitionSpec.
  • Breaking Changes
    • the initial argument to reduction functions like :func:jax.numpy.sum is now required to be a scalar, consistent with the corresponding NumPy API. The previous behavior of broadcating the output against non-scalar initial values was an unintentional implementation detail ({jax-issue}[#14446](https://github.com/google/jax/issues/14446)).

jaxlib 0.4.4 (Feb 16, 2023)

... (truncated)

Commits
  • a002643 Fix stale reference to util.prod.
  • a9421a8 Merge pull request #14728 from mattjj:custom-vjp-bwd-wrapped-fun
  • bf07395 [custom_vjp] bwd function should not be WrappedFun, may run multiple times
  • abc6c9b [sparse] adjust tolerance on bcoo_dot_general_sampled
  • 3abae68 Rollforward of Add a fastpath to pmap_lib for sharding np.ndarray directly in...
  • c73cc49 Use in_shardings and out_shardings since those are the new arguments that pji...
  • a21bdad Merge pull request #14738 from jakevdp:update-sphinx
  • 55d9c06 DOC: update sphinx & sphinx-autodoc-typehints
  • b348fce Merge pull request #14736 from jakevdp:fix-rtd
  • e8a4e64 Merge pull request #14737 from hawkinsp:kaggle
  • 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)

Updates the requirements on [jax](https://github.com/google/jax) to permit the latest version.
- [Release notes](https://github.com/google/jax/releases)
- [Changelog](https://github.com/google/jax/blob/main/CHANGELOG.md)
- [Commits](jax-ml/jax@jax-v0.2.17...jax-v0.4.5)

---
updated-dependencies:
- dependency-name: jax
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
@dependabot dependabot bot added the dependencies Pull requests that update a dependency file label Mar 3, 2023
@dependabot @github
Copy link
Author

dependabot bot commented on behalf of github Mar 10, 2023

Superseded by #73.

@dependabot dependabot bot closed this Mar 10, 2023
@dependabot dependabot bot deleted the dependabot/pip/jax-gt-0.2.16-and-lt-0.4.6 branch March 10, 2023 16:59
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
Projects
None yet
Development

Successfully merging this pull request may close these issues.

0 participants