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 from 0.4.7 to 0.4.8 #405

Merged
merged 1 commit into from
Mar 30, 2023
Merged

Conversation

dependabot[bot]
Copy link
Contributor

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

Bumps jax from 0.4.7 to 0.4.8.

Changelog

Sourced from jax's changelog.

jax 0.4.8

  • Breaking changes

    • A major component of the Cloud TPU runtime has been upgraded. This enables the following new features on Cloud TPU:

      • {func}jax.debug.print, {func}jax.debug.callback, and {func}jax.debug.breakpoint() now work on Cloud TPU
      • Automatic TPU memory defragmentation

      {func}jax.experimental.host_callback is no longer supported on Cloud TPU with the new runtime component. Please file an issue on the JAX issue tracker if the new jax.debug APIs are insufficient for your use case.

      The old runtime component will be available for at least the next three months by setting the environment variable JAX_USE_PJRT_C_API_ON_TPU=false. If you find you need to disable the new runtime for any reason, please let us know on the JAX issue tracker.

  • Changes

    • The minimum jaxlib version has been bumped from 0.4.6 to 0.4.7.
  • Deprecations

    • CUDA 11.4 support has been dropped. JAX GPU wheels only support CUDA 11.8 and CUDA 12. Older CUDA versions may work if jaxlib is built from source.
    • global_arg_shapes argument of pmap only worked with sharded_jit and has been removed from pmap. Please migrate to pjit and remove global_arg_shapes from pmap.

jaxlib 0.4.8

Commits
  • f282c25 Add minimal pyproject.toml specifying build system
  • cfa330b Merge pull request #15283 from JiaYaobo:fix_wald_doc
  • 2d94f76 Merge pull request #15278 from hawkinsp:cudainstall
  • fbc05ee Remove global_arg_shapes from pmap since it was only used for sharded_jit and...
  • a964ae7 Internal Code Change
  • 7200d07 Merge pull request #15286 from hawkinsp:testjobs
  • d9b0f3c Recommend --local_test_jobs in bazel test command line on GPU.
  • 07fc022 Merge pull request #15279 from hawkinsp:versions
  • 3a4d0b3 remove scale in wald docstring
  • 705b5cc Add version constraints to CUDA pip wheel dependencies.
  • 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 dependabot bot added dependencies Pull requests that update a dependency file python Pull requests that update Python code labels Mar 30, 2023
Bumps [jax](https://github.com/google/jax) from 0.4.7 to 0.4.8.
- [Release notes](https://github.com/google/jax/releases)
- [Changelog](https://github.com/google/jax/blob/main/CHANGELOG.md)
- [Commits](google/jax@jax-v0.4.7...jax-v0.4.8)

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

Signed-off-by: dependabot[bot] <support@github.com>
@schneiderfelipe schneiderfelipe merged commit 80aa0a7 into main Mar 30, 2023
@schneiderfelipe schneiderfelipe deleted the dependabot/pip/main/jax-0.4.8 branch March 30, 2023 13:27
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.

None yet

1 participant