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.4.14 to <=0.4.19 in /tests #167

Closed
wants to merge 1 commit into from

Conversation

dependabot[bot]
Copy link

@dependabot dependabot bot commented on behalf of github Oct 25, 2023

Updates the requirements on jax to permit the latest version.

Changelog

Sourced from jax's changelog.

jax 0.4.19 (Oct 19, 2023)

  • New Features

    • Added {obj}jax.typing.DTypeLike, which can be used to annotate objects that are convertible to JAX dtypes.
    • Added jax.numpy.fill_diagonal.
  • Changes

    • JAX now requires SciPy 1.9 or newer.
  • Bug fixes

    • Only process 0 in a multicontroller distributed JAX program will write persistent compilation cache entries. This fixes write contention if the cache is placed on a network filesystem such as GCS.
    • The version check for cusolver and cufft no longer considers the patch versions when determining if the installed version of these libraries is at least as new as the versions against which JAX was built.

jaxlib 0.4.19 (Oct 19, 2023)

  • Changes
    • jaxlib will now always prefer pip-installed NVIDIA CUDA libraries (nvidia-... packages) over any other CUDA installation if they are installed, including installations named in LD_LIBRARY_PATH. If this causes problems and the intent is to use a system-installed CUDA, the fix is to remove the pip installed CUDA library packages.

jax 0.4.18 (Oct 6, 2023)

jaxlib 0.4.18 (Oct 6, 2023)

  • Changes

    • CUDA jaxlibs now depend on the user to install a compatible NCCL version. If using the recommended cuda12_pip installation, NCCL should be installed automatically. Currently, NCCL 2.16 or newer is required.
    • We now provide Linux aarch64 wheels, both with and without NVIDIA GPU support.
  • Deprecations

    • A number of internal utilities and inadvertent exports in {mod}jax.lax have been deprecated, and will be removed in a future release.
      • jax.lax.dtypes: use jax.dtypes instead.
      • jax.lax.itertools: use itertools instead.
      • naryop, naryop_dtype_rule, standard_abstract_eval, standard_naryop, standard_primitive, standard_unop, unop, and unop_dtype_rule are internal utilities, now deprecated without replacement.
  • Bug fixes

    • Fixed Cloud TPU regression where compilation would OOM due to smem.

... (truncated)

Commits
  • 604dc24 Update XLA dependency to use revision
  • f10e333 Start release for jax 0.4.19
  • 3778265 Merge pull request #18126 from niqodea:wrapcauchy
  • 88fe0da Merge pull request #18078 from ROCmSoftwarePlatform:rocm-jax-triton
  • 9435a0a Merge pull request #18138 from mattjj:shmap-axis-env-fix
  • d55085f [Mosaic] Support tpu.concatenate along the tiling dims as long as the shapes ...
  • cf65480 [XlaCallModule] Drop support for dim_args_spec attribute.
  • 890b762 feat: add wrapcauchy logpdf and pdf
  • e7dff2c Update XLA dependency to use revision
  • b4b97cd [ROCm]: Add jax-triton support for ROCm
  • 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 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)

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@jaxlib-v0.1.32...jax-v0.4.19)

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

Signed-off-by: dependabot[bot] <support@github.com>
@dependabot dependabot bot added category: dependency_changes python Pull requests that update Python code labels Oct 25, 2023
Copy link
Author

dependabot bot commented on behalf of github Nov 3, 2023

Superseded by #174.

@dependabot dependabot bot closed this Nov 3, 2023
@dependabot dependabot bot deleted the dependabot/pip/tests/jax-lte-0.4.19 branch November 3, 2023 08:27
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
category: dependency_changes python Pull requests that update Python code
Projects
None yet
Development

Successfully merging this pull request may close these issues.

0 participants