pip prod(deps): bump jax from 0.4.33 to 0.4.34 #3250
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
![dependabot[bot]](https://avatars.githubusercontent.com/in/29110?s=60&v=4){kind=small}
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## main #3250 +/- ##
=======================================
Coverage 89.92% 89.92%
=======================================
Files 441 441
Lines 55417 55417
Branches 8643 8643
=======================================
Hits 49832 49832
Misses 4158 4158
Partials 1427 1427 ☔ View full report in Codecov by Sentry. |
Bumps [jax](https://github.com/jax-ml/jax) from 0.4.33 to 0.4.34. - [Release notes](https://github.com/jax-ml/jax/releases) - [Changelog](https://github.com/jax-ml/jax/blob/main/CHANGELOG.md) - [Commits](jax-ml/jax@jax-v0.4.33...jax-v0.4.34) --- updated-dependencies: - dependency-name: jax dependency-type: direct:production update-type: version-update:semver-patch ... Signed-off-by: dependabot[bot] <support@github.com>
dependabot
bot
force-pushed
the
dependabot/pip/jax-0.4.34
branch
from
6cc080d to
5baa76a
Compare
emmau678
pushed a commit
that referenced
this pull request
Oct 8, 2024
Bumps [jax](https://github.com/jax-ml/jax) from 0.4.33 to 0.4.34. <details> <summary>Release notes</summary> <p><em>Sourced from <a href="https://github.com/jax-ml/jax/releases">jax's releases</a>.</em></p> <blockquote> <h2>JAX v0.4.34</h2> <ul> <li> <p>New Functionality</p> <ul> <li>This release includes wheels for Python 3.13. Free-threading mode is not yet supported.</li> <li><code>jax.errors.JaxRuntimeError</code> has been added as a public alias for the formerly private <code>XlaRuntimeError</code> type.</li> </ul> </li> <li> <p>Breaking changes</p> <ul> <li><code>jax_pmap_no_rank_reduction</code> flag is set to <code>True</code> by default. <ul> <li><code>array[0]</code> on a pmap result now introduces a reshape (use <code>array[0:1]</code> instead).</li> <li>The per-shard shape (accessable via <code>jax_array.addressable_shards</code> or <code>jax_array.addressable_data(0))</code> now has a leading <code>(1, ...)</code>. Update code that directly accesses shards accordingly. The rank of the per-shard-shape now matches that of the global shape which is the same behavior as jit. This avoids costly reshapes when passing results from pmap into jit.</li> </ul> </li> <li><code>jax.experimental.host_callback</code> has been deprecated since March 2024, with JAX version 0.4.26. Now we set the default value of the <code>--jax_host_callback_legacy</code> configuration value to <code>True</code>, which means that if your code uses <code>jax.experimental.host_callback</code> APIs, those API calls will be implemented in terms of the new <code>jax.experimental.io_callback</code> API. If this breaks your code, for a very limited time, you can set the <code>--jax_host_callback_legacy</code> to <code>True</code>. Soon we will remove that configuration option, so you should instead transition to using the new JAX callback APIs. See <a href="https://redirect.github.com/jax-ml/jax/issues/20385">#20385</a> for a discussion.</li> </ul> </li> <li> <p>Deprecations</p> <ul> <li>In <code>jax.numpy.trim_zeros</code>, non-arraylike arguments or arraylike arguments with <code>ndim != 1</code> are now deprecated, and in the future will result in an error.</li> <li>Internal pretty-printing tools <code>jax.core.pp_*</code> have been removed, after being deprecated in JAX v0.4.30.</li> <li><code>jax.lib.xla_client.Device</code> is deprecated; use <code>jax.Device</code> instead.</li> <li><code>jax.lib.xla_client.XlaRuntimeError</code> has been deprecated. Use <code>jax.errors.JaxRuntimeError</code> instead.</li> <li>The default behavior of <code>jax.pure_callback</code> and <code>jax.extend.ffi.ffi_call</code> under <code>vmap</code> has been deprecated and so has the <code>vectorized</code> parameter to those functions. The <code>vmap_method</code> parameter should be used instead for better defined behavior. See the discussion in <a href="https://redirect.github.com/jax-ml/jax/issues/23881">#23881</a> for more details.</li> </ul> </li> <li> <p>Deletion:</p> <ul> <li><code>jax.xla_computation</code> is deleted. It has been 3 months since its deprecation in 0.4.30 JAX release. Please use the AOT APIs to get the same functionality as <code>jax.xla_computation</code>. <ul> <li><code>jax.xla_computation(fn)(*args, **kwargs)</code> can be replaced with <code>jax.jit(fn).lower(*args, **kwargs).compiler_ir('hlo')</code>.</li> <li>You can also use <code>.out_info</code> property of <code>jax.stages.Lowered</code> to get the output information (like tree structure, shape and dtype).</li> <li>For cross-backend lowering, you can replace</li> </ul> </li> </ul> </li> </ul> <!-- raw HTML omitted --> </blockquote> <p>... (truncated)</p> </details> <details> <summary>Changelog</summary> <p><em>Sourced from <a href="https://github.com/jax-ml/jax/blob/main/CHANGELOG.md">jax's changelog</a>.</em></p> <blockquote> <h2>jax 0.4.34 (October 4, 2023)</h2> <ul> <li> <p>New Functionality</p> <ul> <li>This release includes wheels for Python 3.13. Free-threading mode is not yet supported.</li> <li><code>jax.errors.JaxRuntimeError</code> has been added as a public alias for the formerly private <code>XlaRuntimeError</code> type.</li> </ul> </li> <li> <p>Breaking changes</p> <ul> <li><code>jax_pmap_no_rank_reduction</code> flag is set to <code>True</code> by default. <ul> <li>array[0] on a pmap result now introduces a reshape (use array[0:1] instead).</li> <li>The per-shard shape (accessable via jax_array.addressable_shards or jax_array.addressable_data(0)) now has a leading (1, ...). Update code that directly accesses shards accordingly. The rank of the per-shard-shape now matches that of the global shape which is the same behavior as jit. This avoids costly reshapes when passing results from pmap into jit.</li> </ul> </li> <li><code>jax.experimental.host_callback</code> has been deprecated since March 2024, with JAX version 0.4.26. Now we set the default value of the <code>--jax_host_callback_legacy</code> configuration value to <code>True</code>, which means that if your code uses <code>jax.experimental.host_callback</code> APIs, those API calls will be implemented in terms of the new <code>jax.experimental.io_callback</code> API. If this breaks your code, for a very limited time, you can set the <code>--jax_host_callback_legacy</code> to <code>True</code>. Soon we will remove that configuration option, so you should instead transition to using the new JAX callback APIs. See {jax-issue}<code>[#20385](https://github.com/jax-ml/jax/issues/20385)</code> for a discussion.</li> </ul> </li> <li> <p>Deprecations</p> <ul> <li>In {func}<code>jax.numpy.trim_zeros</code>, non-arraylike arguments or arraylike arguments with <code>ndim != 1</code> are now deprecated, and in the future will result in an error.</li> <li>Internal pretty-printing tools <code>jax.core.pp_*</code> have been removed, after being deprecated in JAX v0.4.30.</li> <li><code>jax.lib.xla_client.Device</code> is deprecated; use <code>jax.Device</code> instead.</li> <li><code>jax.lib.xla_client.XlaRuntimeError</code> has been deprecated. Use <code>jax.errors.JaxRuntimeError</code> instead.</li> <li>The default behavior of {func}<code>jax.pure_callback</code> and {func}<code>jax.extend.ffi.ffi_call</code> under <code>vmap</code> has been deprecated and so has the <code>vectorized</code> parameter to those functions. The <code>vmap_method</code> parameter should be used instead for better defined behavior. See the discussion in {jax-issue}<code>[#23881](https://github.com/jax-ml/jax/issues/23881)</code> for more details.</li> </ul> </li> <li> <p>Deletion:</p> <ul> <li><code>jax.xla_computation</code> is deleted. It's been 3 months since it's deprecation in 0.4.30 JAX release. Please use the AOT APIs to get the same functionality as <code>jax.xla_computation</code>. <ul> <li><code>jax.xla_computation(fn)(*args, **kwargs)</code> can be replaced with <code>jax.jit(fn).lower(*args, **kwargs).compiler_ir('hlo')</code>.</li> <li>You can also use <code>.out_info</code> property of <code>jax.stages.Lowered</code> to get the output information (like tree structure, shape and dtype).</li> </ul> </li> </ul> </li> </ul> <!-- raw HTML omitted --> </blockquote> <p>... (truncated)</p> </details> <details> <summary>Commits</summary> <ul> <li><a href="https://github.com/jax-ml/jax/commit/affba367c5533df8900e32cbc3d31ca92dd1c1ea"><code>affba36</code></a> JAX 0.4.34 release</li> <li><a href="https://github.com/jax-ml/jax/commit/59cb6927e44f8bd21567f3e195f406f8b5019817"><code>59cb692</code></a> Update XLA dependency to use revision</li> <li><a href="https://github.com/jax-ml/jax/commit/8fc808222502f48d65cbce79ec521f9b1903ad01"><code>8fc8082</code></a> Merge pull request <a href="https://redirect.github.com/jax-ml/jax/issues/24077">#24077</a> from jakevdp:badge-links</li> <li><a href="https://github.com/jax-ml/jax/commit/cfb75411c8d14811120d7a20ea28e8050ddbd492"><code>cfb7541</code></a> Merge pull request <a href="https://redirect.github.com/jax-ml/jax/issues/23738">#23738</a> from keshavb96:disable_remat_pass</li> <li><a href="https://github.com/jax-ml/jax/commit/78b65dd05a5b82f79b1b19661b782ec581fc344b"><code>78b65dd</code></a> Merge pull request <a href="https://redirect.github.com/jax-ml/jax/issues/24070">#24070</a> from jakevdp:issubdtype-doc</li> <li><a href="https://github.com/jax-ml/jax/commit/a8744e38d30fc99b1e44256c7a8877d8be0ccdc2"><code>a8744e3</code></a> README: fix badge href links</li> <li><a href="https://github.com/jax-ml/jax/commit/152a873184d55ec20f4d1d833cf45417674a4f9e"><code>152a873</code></a> Update XLA dependency to use revision</li> <li><a href="https://github.com/jax-ml/jax/commit/c9f946ef571f71d8ed30d84b16274c1fcc75671b"><code>c9f946e</code></a> Only thread a discharged ref value through a cond when it changes in some branch</li> <li><a href="https://github.com/jax-ml/jax/commit/816947b6566c9c759fd436fe04b188f41e8ba9a6"><code>816947b</code></a> Merge pull request <a href="https://redirect.github.com/jax-ml/jax/issues/24071">#24071</a> from hawkinsp:winci</li> <li><a href="https://github.com/jax-ml/jax/commit/e212c7733654d5b01c9485294e04cb0fe4840bd2"><code>e212c77</code></a> Merge pull request <a href="https://redirect.github.com/jax-ml/jax/issues/23891">#23891</a> from ROCm:build-fixes-rollup</li> <li>Additional commits viewable in <a href="https://github.com/jax-ml/jax/compare/jax-v0.4.33...jax-v0.4.34">compare view</a></li> </ul> </details> <br /> [](https://docs.github.com/en/github/managing-security-vulnerabilities/about-dependabot-security-updates#about-compatibility-scores) 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-automerge-start) [//]: # (dependabot-automerge-end) --- <details> <summary>Dependabot commands and options</summary> <br /> 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) </details> Signed-off-by: dependabot[bot] <support@github.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Bumps jax from 0.4.33 to 0.4.34.
Release notes
Sourced from jax's releases.
... (truncated)
Changelog
Sourced from jax's changelog.
... (truncated)
Commits
affba36JAX 0.4.34 release59cb692Update XLA dependency to use revision8fc8082Merge pull request #24077 from jakevdp:badge-linkscfb7541Merge pull request #23738 from keshavb96:disable_remat_pass78b65ddMerge pull request #24070 from jakevdp:issubdtype-doca8744e3README: fix badge href links152a873Update XLA dependency to use revisionc9f946eOnly thread a discharged ref value through a cond when it changes in some branch816947bMerge pull request #24071 from hawkinsp:wincie212c77Merge pull request #23891 from ROCm:build-fixes-rollupDependabot 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 rebasewill rebase this PR@dependabot recreatewill recreate this PR, overwriting any edits that have been made to it@dependabot mergewill merge this PR after your CI passes on it@dependabot squash and mergewill squash and merge this PR after your CI passes on it@dependabot cancel mergewill cancel a previously requested merge and block automerging@dependabot reopenwill reopen this PR if it is closed@dependabot closewill close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually@dependabot show <dependency name> ignore conditionswill show all of the ignore conditions of the specified dependency@dependabot ignore this major versionwill 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 versionwill 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 dependencywill close this PR and stop Dependabot creating any more for this dependency (unless you reopen the PR or upgrade to it yourself)