Skip to content

Bump jax from 0.4.16 to 0.4.17 in /mpi4jax/_src #217

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

Merged
merged 1 commit into from
Oct 4, 2023

Conversation

dependabot[bot]
Copy link
Contributor

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

Bumps jax from 0.4.16 to 0.4.17.

Changelog

Sourced from jax's changelog.

jax 0.4.17 (Oct 3, 2023)

  • New features
    • Added new {func}jax.numpy.bitwise_count function, matching the API of the simlar function recently added to NumPy.
  • Deprecations
    • Removed the deprecated module jax.abstract_arrays and all its contents.
    • Named key constructors in {mod}jax.random are deprecated. Pass the impl argument to {func}jax.random.PRNGKey or {func}jax.random.key instead:
      • random.threefry2x32_key(seed) becomes random.PRNGKey(seed, impl='threefry2x32')
      • random.rbg_key(seed) becomes random.PRNGKey(seed, impl='rbg')
      • random.unsafe_rbg_key(seed) becomes random.PRNGKey(seed, impl='unsafe_rbg')
  • Changes:
    • CUDA: JAX now verifies that the CUDA libraries it finds are at least as new as the CUDA libraries that JAX was built against. If older libraries are found, JAX raises an exception since that is preferable to mysterious failures and crashes.
    • Removed the "No GPU/TPU" found warning. Instead warn if, on Linux, an NVIDIA GPU or a Google TPU are found but not used and --jax_platforms was not specified.
    • {func}jax.scipy.stats.mode now returns a 0 count if the mode is taken across a size-0 axis, matching the behavior of scipy.stats.mode in SciPy 1.11.
    • Most jax.numpy functions and attributes now have fully-defined type stubs. Previously many of these were treated as Any by static type checkers like mypy and pytype.

jaxlib 0.4.17 (Oct 3, 2023)

  • Changes:

    • Python 3.12 wheels were added in this release.
    • The CUDA 12 wheels now require CUDA 12.2 or newer and cuDNN 8.9.4 or newer.
  • Bug fixes:

    • Fixed log spam from ABSL when the JAX CPU backend was initialized.
Commits
  • 4cb5eee Merge pull request #17905 from skye:version
  • b9c602b Update XLA commit and versions for jax 0.4.17 release
  • 3d84808 Merge pull request #17851 from mattjj:readme-tweak
  • dddbe43 Update README.md
  • 24ad445 [Pallas] Add support for pytrees in scalar prefetch
  • ee8af09 Fix mock_gpu_test on OSS build.
  • c3e73c6 Merge pull request #17760 from superbobry:array-any
  • b2ac2de [XLA] Split --xla_detailed_logging_and_dumping debug flag into --xla_detailed...
  • 72407f6 Update XLA dependency to use revision
  • 17e259b fix typo: device_fun(c) -> device_fun(x)
  • 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](https://github.com/google/jax) from 0.4.16 to 0.4.17.
- [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.16...jax-v0.4.17)

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

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

codecov bot commented Oct 4, 2023

Codecov Report

All modified lines are covered by tests ✅

Comparison is base (3cf1d6d) 65.92% compared to head (16d7c28) 65.92%.
Report is 1 commits behind head on master.

Additional details and impacted files
@@           Coverage Diff           @@
##           master     #217   +/-   ##
=======================================
  Coverage   65.92%   65.92%           
=======================================
  Files          24       24           
  Lines        1265     1265           
  Branches      118      118           
=======================================
  Hits          834      834           
  Misses        401      401           
  Partials       30       30           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

@PhilipVinc PhilipVinc merged commit 740cf63 into master Oct 4, 2023
@PhilipVinc PhilipVinc deleted the dependabot/pip/mpi4jax/_src/jax-0.4.17 branch October 4, 2023 14:33
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.

1 participant