pip prod(deps): bump jax from 0.4.35 to 0.4.36 #3577
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.35 to 0.4.36. - [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.35...jax-v0.4.36) --- updated-dependencies: - dependency-name: jax dependency-type: direct:production update-type: version-update:semver-patch ... Signed-off-by: dependabot[bot] <support@github.com>
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## main #3577 +/- ##
=======================================
Coverage 90.40% 90.40%
=======================================
Files 469 469
Lines 58941 58941
Branches 5611 5611
=======================================
Hits 53287 53287
Misses 4213 4213
Partials 1441 1441 ☔ View full report in Codecov by Sentry. |
EdmundGoodman
pushed a commit
to EdmundGoodman/xdsl
that referenced
this pull request
Dec 6, 2024
Bumps [jax](https://github.com/jax-ml/jax) from 0.4.35 to 0.4.36. <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.36</h2> <ul> <li>Breaking Changes <ul> <li> <p>This release lands "stackless", an internal change to JAX's tracing machinery. We made trace dispatch purely a function of context rather than a function of both context and data. This let us delete a lot of machinery for managing data-dependent tracing: levels, sublevels, <code>post_process_call</code>, <code>new_base_main</code>, <code>custom_bind</code>, and so on. The change should only affect users that use JAX internals.</p> <p>If you do use JAX internals then you may need to update your code (see <a href="https://github.com/jax-ml/jax/commit/c36e1f7c1ad4782060cbc8e8c596d85dfb83986f">https://github.com/jax-ml/jax/commit/c36e1f7c1ad4782060cbc8e8c596d85dfb83986f</a> for clues about how to do this). There might also be version skew issues with JAX libraries that do this. If you find this change breaks your non-JAX-internals-using code then try the <code>config.jax_data_dependent_tracing_fallback</code> flag as a workaround, and if you need help updating your code then please file a bug.</p> </li> <li> <p><code>jax.experimental.jax2tf.convert</code> with <code>native_serialization=False</code> or with <code>enable_xla=False</code> have been deprecated since July 2024, with JAX version 0.4.31. Now we removed support for these use cases. <code>jax2tf</code> with native serialization will still be supported.</p> </li> <li> <p>In <code>jax.interpreters.xla</code>, the <code>xb</code>, <code>xc</code>, and <code>xe</code> symbols have been removed after being deprecated in JAX v0.4.31. Instead use <code>xb = jax.lib.xla_bridge</code>, <code>xc = jax.lib.xla_client</code>, and <code>xe = jax.lib.xla_extension</code>.</p> </li> <li> <p>The deprecated module <code>jax.experimental.export</code> has been removed. It was replaced by <code>jax.export</code> in JAX v0.4.30. See the <a href="https://jax.readthedocs.io/en/latest/export/export.html#migration-guide-from-jax-experimental-export">migration guide</a> for information on migrating to the new API.</p> </li> <li> <p>The <code>initial</code> argument to <code>jax.nn.softmax</code> and <code>jax.nn.log_softmax</code> has been removed, after being deprecated in v0.4.27.</p> </li> <li> <p>Calling <code>np.asarray</code> on typed PRNG keys (i.e. keys produced by <code>jax.random.key</code>) now raises an error. Previously, this returned a scalar object array.</p> </li> <li> <p>The following deprecated methods and functions in <code>jax.export</code> have been removed:</p> <ul> <li><code>jax.export.DisabledSafetyCheck.shape_assertions</code>: it had no effect already.</li> <li><code>jax.export.Exported.lowering_platforms</code>: use <code>platforms</code>.</li> <li><code>jax.export.Exported.mlir_module_serialization_version</code>: use <code>calling_convention_version</code>.</li> <li><code>jax.export.Exported.uses_shape_polymorphism</code>: use <code>uses_global_constants</code>.</li> <li>the <code>lowering_platforms</code> kwarg for <code>jax.export.export</code>: use <code>platforms</code> instead.</li> </ul> </li> <li> <p>The kwargs <code>symbolic_scope</code> and <code>symbolic_constraints</code> from <code>jax.export.symbolic_args_specs</code> have been removed. They were deprecated in June 2024. Use <code>scope</code> and <code>constraints</code> instead.</p> </li> <li> <p>Hashing of tracers, which has been deprecated since version 0.4.30, now results in a <code>TypeError</code>.</p> </li> <li> <p>Refactor: JAX build CLI (build/build.py) now uses a subcommand structure and replaces previous build.py usage. Run <code>python build/build.py --help</code> for more details. Brief overview of the new subcommand options:</p> </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.36</h2> <ul> <li>Breaking Changes <ul> <li> <p>This release lands "stackless", an internal change to JAX's tracing machinery. We made trace dispatch purely a function of context rather than a function of both context and data. This let us delete a lot of machinery for managing data-dependent tracing: levels, sublevels, <code>post_process_call</code>, <code>new_base_main</code>, <code>custom_bind</code>, and so on. The change should only affect users that use JAX internals.</p> <p>If you do use JAX internals then you may need to update your code (see <a href="https://github.com/jax-ml/jax/commit/c36e1f7c1ad4782060cbc8e8c596d85dfb83986f">https://github.com/jax-ml/jax/commit/c36e1f7c1ad4782060cbc8e8c596d85dfb83986f</a> for clues about how to do this). There might also be version skew issues with JAX libraries that do this. If you find this change breaks your non-JAX-internals-using code then try the <code>config.jax_data_dependent_tracing_fallback</code> flag as a workaround, and if you need help updating your code then please file a bug.</p> </li> <li> <p>{func}<code>jax.experimental.jax2tf.convert</code> with <code>native_serialization=False</code> or with <code>enable_xla=False</code> have been deprecated since July 2024, with JAX version 0.4.31. Now we removed support for these use cases. <code>jax2tf</code> with native serialization will still be supported.</p> </li> <li> <p>In <code>jax.interpreters.xla</code>, the <code>xb</code>, <code>xc</code>, and <code>xe</code> symbols have been removed after being deprecated in JAX v0.4.31. Instead use <code>xb = jax.lib.xla_bridge</code>, <code>xc = jax.lib.xla_client</code>, and <code>xe = jax.lib.xla_extension</code>.</p> </li> <li> <p>The deprecated module <code>jax.experimental.export</code> has been removed. It was replaced by {mod}<code>jax.export</code> in JAX v0.4.30. See the <a href="https://jax.readthedocs.io/en/latest/export/export.html#migration-guide-from-jax-experimental-export">migration guide</a> for information on migrating to the new API.</p> </li> <li> <p>The <code>initial</code> argument to {func}<code>jax.nn.softmax</code> and {func}<code>jax.nn.log_softmax</code> has been removed, after being deprecated in v0.4.27.</p> </li> <li> <p>Calling <code>np.asarray</code> on typed PRNG keys (i.e. keys produced by :func:<code>jax.random.key</code>) now raises an error. Previously, this returned a scalar object array.</p> </li> <li> <p>The following deprecated methods and functions in {mod}<code>jax.export</code> have been removed:</p> <ul> <li><code>jax.export.DisabledSafetyCheck.shape_assertions</code>: it had no effect already.</li> <li><code>jax.export.Exported.lowering_platforms</code>: use <code>platforms</code>.</li> <li><code>jax.export.Exported.mlir_module_serialization_version</code>: use <code>calling_convention_version</code>.</li> <li><code>jax.export.Exported.uses_shape_polymorphism</code>: use <code>uses_global_constants</code>.</li> <li>the <code>lowering_platforms</code> kwarg for {func}<code>jax.export.export</code>: use <code>platforms</code> instead.</li> </ul> </li> <li> <p>The kwargs <code>symbolic_scope</code> and <code>symbolic_constraints</code> from {func}<code>jax.export.symbolic_args_specs</code> have been removed. They were deprecated in June 2024. Use <code>scope</code> and <code>constraints</code> instead.</p> </li> <li> <p>Hashing of tracers, which has been deprecated since version 0.4.30, now results in a <code>TypeError</code>.</p> </li> <li> <p>Refactor: JAX build CLI (build/build.py) now uses a subcommand structure and replaces previous build.py usage. Run <code>python build/build.py --help</code> for</p> </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/259194a69f52a06847a9ff11eb268072e91fd65f"><code>259194a</code></a> [Pallas] Fix shard_axis in dma_start interpret mode rule.</li> <li><a href="https://github.com/jax-ml/jax/commit/7e6620a57775084dfa8d438ae4fd27f3ef365018"><code>7e6620a</code></a> JAX release 0.4.36.</li> <li><a href="https://github.com/jax-ml/jax/commit/23d5c10ff0704f66ad7ec65a8cdcd09bd2420591"><code>23d5c10</code></a> [Mosaic:TPU] Fix fully replicated relayout</li> <li><a href="https://github.com/jax-ml/jax/commit/2a4a0e8d6fb36b59f9c6f24e0018d42c8c8d8ee9"><code>2a4a0e8</code></a> [jax:custom_partitioning] Implement SdyShardingRule to support</li> <li><a href="https://github.com/jax-ml/jax/commit/f73fa7a7ad64b2f15e8669beed14600704287b93"><code>f73fa7a</code></a> Merge pull request <a href="https://redirect.github.com/jax-ml/jax/issues/25290">#25290</a> from jakevdp:reduction-where</li> <li><a href="https://github.com/jax-ml/jax/commit/a71f9a62e6f67640a4b0578d042b07792fcf407a"><code>a71f9a6</code></a> Merge pull request <a href="https://redirect.github.com/jax-ml/jax/issues/25271">#25271</a> from jakevdp:fix-vector-norm</li> <li><a href="https://github.com/jax-ml/jax/commit/e20a483befbb80bbf782b931ec57a44c78c313b8"><code>e20a483</code></a> [JAX] Add end-to-end execution support in colocated Python API</li> <li><a href="https://github.com/jax-ml/jax/commit/aaaee63ac5887a8f7cf06655a42053eb9c06b91b"><code>aaaee63</code></a> jnp.linalg.vector_norm: properly support multiple axes</li> <li><a href="https://github.com/jax-ml/jax/commit/29a8cce66cfe35f216b958779e975910d280bb5b"><code>29a8cce</code></a> jax.numpy: require boolean dtype for where argument</li> <li><a href="https://github.com/jax-ml/jax/commit/3f5f3e1c47c230cc5d44841b08c3db9598442d13"><code>3f5f3e1</code></a> [export] Removed __gpu$xla.gpu.triton (Pallas GPU) from the list of custom ca...</li> <li>Additional commits viewable in <a href="https://github.com/jax-ml/jax/compare/jax-v0.4.35...jax-v0.4.36">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.35 to 0.4.36.
Release notes
Sourced from jax's releases.
... (truncated)
Changelog
Sourced from jax's changelog.
... (truncated)
Commits
259194a[Pallas] Fix shard_axis in dma_start interpret mode rule.7e6620aJAX release 0.4.36.23d5c10[Mosaic:TPU] Fix fully replicated relayout2a4a0e8[jax:custom_partitioning] Implement SdyShardingRule to supportf73fa7aMerge pull request #25290 from jakevdp:reduction-wherea71f9a6Merge pull request #25271 from jakevdp:fix-vector-norme20a483[JAX] Add end-to-end execution support in colocated Python APIaaaee63jnp.linalg.vector_norm: properly support multiple axes29a8ccejax.numpy: require boolean dtype for where argument3f5f3e1[export] Removed __gpu$xla.gpu.triton (Pallas GPU) from the list of custom ca...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)