pip prod(deps): bump jax from 0.4.38 to 0.5.0 #3764
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}
Bumps [jax](https://github.com/jax-ml/jax) from 0.4.38 to 0.5.0. - [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.38...jax-v0.5.0) --- updated-dependencies: - dependency-name: jax dependency-type: direct:production update-type: version-update:semver-minor ... Signed-off-by: dependabot[bot] <support@github.com>
dependabot
bot
added
the
dependencies
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## main #3764 +/- ##
==========================================
- Coverage 91.23% 91.23% -0.01%
==========================================
Files 459 459
Lines 57307 57344 +37
Branches 5532 5530 -2
==========================================
+ Hits 52286 52317 +31
- Misses 3594 3604 +10
+ Partials 1427 1423 -4 ☔ View full report in Codecov by Sentry. |
|
OK, I won't notify you again about this release, but will get in touch when a new version is available. If you'd rather skip all updates until the next major or minor version, let me know by commenting If you change your mind, just re-open this PR and I'll resolve any conflicts on it. |
oluwatimilehin
pushed a commit
to oluwatimilehin/xdsl
that referenced
this pull request
Feb 13, 2025
Bumps [jax](https://github.com/jax-ml/jax) from 0.4.38 to 0.5.0. <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.5.0</h2> <p>As of this release, JAX now uses <a href="https://jax.readthedocs.io/en/latest/jep/25516-effver.html">effort-based versioning</a>. Since this release makes a breaking change to PRNG key semantics that may require users to update their code, we are bumping the "meso" version of JAX to signify this.</p> <ul> <li> <p>Breaking changes</p> <ul> <li> <p>Enable <code>jax_threefry_partitionable</code> by default (see <a href="https://github.com/jax-ml/jax/discussions/18480">the update note</a>).</p> </li> <li> <p>This release drops support for Mac x86 wheels. Mac ARM of course remains supported. For a recent discussion, see <a href="https://github.com/jax-ml/jax/discussions/22936">https://github.com/jax-ml/jax/discussions/22936</a>.</p> <p>Two key factors motivated this decision:</p> <ul> <li>The Mac x86 build (only) has a number of test failures and crashes. We would prefer to ship no release than a broken release.</li> <li>Mac x86 hardware is end-of-life and cannot be easily obtained for developers at this point. So it is difficult for us to fix this kind of problem even if we wanted to.</li> </ul> <p>We are open to readding support for Mac x86 if the community is willing to help support that platform: in particular, we would need the JAX test suite to pass cleanly on Mac x86 before we could ship releases again.</p> </li> </ul> </li> <li> <p>Changes:</p> <ul> <li>The minimum NumPy version is now 1.25. NumPy 1.25 will remain the minimum supported version until June 2025.</li> <li>The minimum SciPy version is now 1.11. SciPy 1.11 will remain the minimum supported version until June 2025.</li> <li><code>jax.numpy.einsum</code> now defaults to <code>optimize='auto'</code> rather than <code>optimize='optimal'</code>. This avoids exponentially-scaling trace-time in the case of many arguments (<code>[#25214](https://github.com/jax-ml/jax/issues/25214)</code>).</li> <li><code>jax.numpy.linalg.solve</code> no longer supports batched 1D arguments on the right hand side. To recover the previous behavior in these cases, use <code>solve(a, b[..., None]).squeeze(-1)</code>.</li> </ul> </li> <li> <p>New Features</p> <ul> <li><code>jax.numpy.fft.fftn</code>, <code>jax.numpy.fft.rfftn</code>, <code>jax.numpy.fft.ifftn</code>, and <code>jax.numpy.fft.irfftn</code> now support transforms in more than 3 dimensions, which was previously the limit. See <code>[#25606](https://github.com/jax-ml/jax/issues/25606)</code> for more details.</li> <li>Support added for user defined state in the FFI via the new <code>jax.ffi.register_ffi_type_id</code> function.</li> <li>The AOT lowering <code>.as_text()</code> method now supports the <code>debug_info</code> option to include debugging information, e.g., source location, in the output.</li> </ul> </li> <li> <p>Deprecations</p> <ul> <li>From <code>jax.interpreters.xla</code>, <code>abstractify</code> and <code>pytype_aval_mappings</code> are now deprecated, having been replaced by symbols of the same name in <code>jax.core</code>.</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.5.0 (Jan 17, 2025)</h2> <p>As of this release, JAX now uses <a href="https://jax.readthedocs.io/en/latest/jep/25516-effver.html">effort-based versioning</a>. Since this release makes a breaking change to PRNG key semantics that may require users to update their code, we are bumping the "meso" version of JAX to signify this.</p> <ul> <li> <p>Breaking changes</p> <ul> <li> <p>Enable <code>jax_threefry_partitionable</code> by default (see <a href="https://github.com/jax-ml/jax/discussions/18480">the update note</a>).</p> </li> <li> <p>This release drops support for Mac x86 wheels. Mac ARM of course remains supported. For a recent discussion, see <a href="https://github.com/jax-ml/jax/discussions/22936">https://github.com/jax-ml/jax/discussions/22936</a>.</p> <p>Two key factors motivated this decision:</p> <ul> <li>The Mac x86 build (only) has a number of test failures and crashes. We would prefer to ship no release than a broken release.</li> <li>Mac x86 hardware is end-of-life and cannot be easily obtained for developers at this point. So it is difficult for us to fix this kind of problem even if we wanted to.</li> </ul> <p>We are open to readding support for Mac x86 if the community is willing to help support that platform: in particular, we would need the JAX test suite to pass cleanly on Mac x86 before we could ship releases again.</p> </li> </ul> </li> <li> <p>Changes:</p> <ul> <li>The minimum NumPy version is now 1.25. NumPy 1.25 will remain the minimum supported version until June 2025.</li> <li>The minimum SciPy version is now 1.11. SciPy 1.11 will remain the minimum supported version until June 2025.</li> <li>{func}<code>jax.numpy.einsum</code> now defaults to <code>optimize='auto'</code> rather than <code>optimize='optimal'</code>. This avoids exponentially-scaling trace-time in the case of many arguments ({jax-issue}<code>[#25214](https://github.com/jax-ml/jax/issues/25214)</code>).</li> <li>{func}<code>jax.numpy.linalg.solve</code> no longer supports batched 1D arguments on the right hand side. To recover the previous behavior in these cases, use <code>solve(a, b[..., None]).squeeze(-1)</code>.</li> </ul> </li> <li> <p>New Features</p> <ul> <li>{func}<code>jax.numpy.fft.fftn</code>, {func}<code>jax.numpy.fft.rfftn</code>, {func}<code>jax.numpy.fft.ifftn</code>, and {func}<code>jax.numpy.fft.irfftn</code> now support transforms in more than 3 dimensions, which was previously the limit. See {jax-issue}<code>[#25606](https://github.com/jax-ml/jax/issues/25606)</code> for more details.</li> <li>Support added for user defined state in the FFI via the new {func}<code>jax.ffi.register_ffi_type_id</code> function.</li> <li>The AOT lowering <code>.as_text()</code> method now supports the <code>debug_info</code> option to include debugging information, e.g., source location, in the output.</li> </ul> </li> <li> <p>Deprecations</p> </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/c25fb92c44d0961f9bef0edaf0cf294d120772ee"><code>c25fb92</code></a> Release JAX 0.5.0</li> <li><a href="https://github.com/jax-ml/jax/commit/a527aba646260c49fd88f421b911296bf5e289ff"><code>a527aba</code></a> Reverts f1b894d14a28ac22a037fb79177b991275c75a18</li> <li><a href="https://github.com/jax-ml/jax/commit/ce85b8988403bdb7c750e60a173dbf5e4cbfc011"><code>ce85b89</code></a> [sharding_in_types] Error out for reshape for splits like this: <code>(4, 6, 8)</code> -...</li> <li><a href="https://github.com/jax-ml/jax/commit/7cac76d346c3f8862af8af473344f2ba69f7e67b"><code>7cac76d</code></a> Update XLA dependency to use revision</li> <li><a href="https://github.com/jax-ml/jax/commit/d3be190efb58c9c5ec9c0263ad3b20f66b771dd1"><code>d3be190</code></a> [Mosaic GPU] Delete unused declarations of <code>mosaic_gpu_memcpy_async_h2d</code>.</li> <li><a href="https://github.com/jax-ml/jax/commit/d34c40f6b6371bf1a41e08e047d7138484931421"><code>d34c40f</code></a> [mosaic_gpu] Added a serialization pass</li> <li><a href="https://github.com/jax-ml/jax/commit/af667199dbc1188d1ee0f70cf8922cdb7d64a701"><code>af66719</code></a> [sharding_in_types] Rename <code>.at[...].get(out_spec)</code> to `.at[...].get(out_shar...</li> <li><a href="https://github.com/jax-ml/jax/commit/97cd748376fe1f26ff0cd8fd9d4d752bed6ce422"><code>97cd748</code></a> Rename out_type -> out_sharding parameter on einsum</li> <li><a href="https://github.com/jax-ml/jax/commit/49224d6cdb4dbae2a8dcc777c295ce6757d5a475"><code>49224d6</code></a> Replace Auto/User/Collective AxisTypes names with Hidden/Visible/Collective.</li> <li><a href="https://github.com/jax-ml/jax/commit/bd22bfef7135c2c03829111b82ff67605760aee4"><code>bd22bfe</code></a> [Mosaic TPU] Use large to compact 2nd minor retiling for conversions going bo...</li> <li>Additional commits viewable in <a href="https://github.com/jax-ml/jax/compare/jax-v0.4.38...jax-v0.5.0">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> Co-authored-by: superlopuh <superlopuh@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.38 to 0.5.0.
Release notes
Sourced from jax's releases.
... (truncated)
Changelog
Sourced from jax's changelog.
... (truncated)
Commits
c25fb92Release JAX 0.5.0a527abaReverts f1b894d14a28ac22a037fb79177b991275c75a18ce85b89[sharding_in_types] Error out for reshape for splits like this:(4, 6, 8)-...7cac76dUpdate XLA dependency to use revisiond3be190[Mosaic GPU] Delete unused declarations ofmosaic_gpu_memcpy_async_h2d.d34c40f[mosaic_gpu] Added a serialization passaf66719[sharding_in_types] Rename.at[...].get(out_spec)to `.at[...].get(out_shar...97cd748Rename out_type -> out_sharding parameter on einsum49224d6Replace Auto/User/Collective AxisTypes names with Hidden/Visible/Collective.bd22bfe[Mosaic TPU] Use large to compact 2nd minor retiling for conversions going bo...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 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)