Skip to content

Conversation

renovate[bot]
Copy link
Contributor

@renovate renovate bot commented Jul 13, 2024

Coming soon: The Renovate bot (GitHub App) will be renamed to Mend. PRs from Renovate will soon appear from 'Mend'. Learn more here.

This PR contains the following updates:

Package Change Age Confidence
flax 0.8.1 -> 0.11.2 age confidence

Warning

Some dependencies could not be looked up. Check the Dependency Dashboard for more information.


Release Notes

google/flax (flax)

v0.11.2: 0.11.2

Compare Source

What's Changed

nnx.merge now doesn't create a copy of the Variables in the incoming states by default, meaning that the new merged structures holds references to the incoming Variables. This enables new patterns, for example its now possible to create models with the same state but with different runtime behavior:

model = SomeModel(...)

### create eval model
eval_model = nnx.merge(*nnx.split(model))  # same Variables, different structure
eval_model.eval()

model and eval_model share the same Variables and are therefore kept in sync but have different runtime behavior, this avoids having to constantly mutate a single model back and forth between different runtime modes which can be error prone / cause unwanted recompilation.

To keep the old behavior use nnx.merge(..., copy=True).

PRs

New Contributors

Full Changelog: google/flax@v0.11.1...v0.11.2

v0.11.1

Compare Source

What's Changed

New Contributors

Full Changelog: google/flax@v0.11.0...v0.11.1

v0.11.0

Compare Source

v0.11.0 - Pytrees, MutableArrays, and more!

This version of Flax introduces some changes to improve interop with native JAX and adds support for the new jax.experimental.MutableArray. More on this soon! However, some breaking changes to align with the JAX way of doing things were necessary. Most code should remain intact, however, the following changes deviate from the current behavior:

  • Rngs in standard layers: all standard layers no longer hold a shared reference to the rngs object given in the constructor, instead they now keep a fork-ed copy of the Rngs or RngStream objects. This impacts Using Rngs in NNX Transforms and Loading Checkpoints with RNGs.
  • Optimizer Updates: the Optimizer abstraction no longer holds a reference to the model to avoid reference sharing, instead the model must be provided as the first argument to update.
  • Modules as Pytrees: Modules are now pytrees! This avoid unnecessary use of split and merge when interacting trivially with raw JAX transforms (state must still be manually propagated if not using MutableArrays, and referential transparency is still an issue). This affects when operating on Pytrees containing NNX Objects with jax.tree.* APIs.

Checkout the full NNX 0.10 to NNX 0.11 migration guide.

In the near future we'll share more information about new ways of using NNX with JAX transforms directly by leveraging the new Pytree and MutableArray support. Stay tuned!

What's Changed

New Contributors

Full Changelog: google/flax@v0.10.7...v0.11.0

v0.10.7: 0.10.7

Compare Source

What's Changed

New Contributors

Full Changelog: google/flax@v0.10.6...v0.10.7

v0.10.6: 0.10.6

Compare Source

What's Changed

Full Changelog: google/flax@v0.10.5...v0.10.6

v0.10.5: 0.10.5

Compare Source

What's Changed

New Contributors

Full Changelog: google/flax@v0.10.4...v0.10.5

v0.10.4: Release 0.10.4

Compare Source

What's Changed

New Contributors

Full Changelog: google/flax@v0.10.3...v0.10.4

v0.10.3: Version 0.10.3

Compare Source

What's Changed

New Contributors

Full Changelog: google/flax@v0.10.2...v0.10.3

v0.10.2: Version 0.10.2

Compare Source

What's Changed

New Contributors

Full Changelog: google/flax@v0.10.1...v0.10.2

v0.10.1: Version 0.10.1

Compare Source

What's Changed

New Contributors

Full Changelog: google/flax@v0.10.0...v0.10.1

v0.10.0: Version 0.10.0

Compare Source

What's Changed


Configuration

📅 Schedule: Branch creation - At any time (no schedule defined), Automerge - At any time (no schedule defined).

🚦 Automerge: Disabled by config. Please merge this manually once you are satisfied.

Rebasing: Whenever PR becomes conflicted, or you tick the rebase/retry checkbox.

🔕 Ignore: Close this PR and you won't be reminded about this update again.


  • If you want to rebase/retry this PR, check this box

This PR was generated by Mend Renovate. View the repository job log.

@renovate renovate bot requested a review from a team as a code owner July 13, 2024 23:35
@renovate renovate bot added the dependencies Pull requests that update a dependency file label Jul 13, 2024
@renovate renovate bot force-pushed the renovate/flax-0.x-lockfile branch from cafeda6 to e803655 Compare August 16, 2024 13:51
@renovate renovate bot force-pushed the renovate/flax-0.x-lockfile branch from e803655 to 1222748 Compare August 27, 2024 19:56
@renovate renovate bot changed the title chore(deps): update dependency flax to v0.8.5 chore(deps): update dependency flax to v0.9.0 Aug 27, 2024
@renovate renovate bot force-pushed the renovate/flax-0.x-lockfile branch from 1222748 to b2e5f65 Compare October 17, 2024 00:23
@renovate renovate bot changed the title chore(deps): update dependency flax to v0.9.0 chore(deps): update dependency flax to v0.10.0 Oct 17, 2024
@renovate renovate bot force-pushed the renovate/flax-0.x-lockfile branch from b2e5f65 to a4243cb Compare November 1, 2024 00:10
@renovate renovate bot changed the title chore(deps): update dependency flax to v0.10.0 chore(deps): update dependency flax to v0.10.1 Nov 1, 2024
@renovate renovate bot force-pushed the renovate/flax-0.x-lockfile branch from a4243cb to 0d48a84 Compare November 19, 2024 21:09
@renovate renovate bot changed the title chore(deps): update dependency flax to v0.10.1 chore(deps): update dependency flax to v0.10.2 Nov 19, 2024
@renovate renovate bot force-pushed the renovate/flax-0.x-lockfile branch from 0d48a84 to 933ac1a Compare February 10, 2025 18:13
@renovate renovate bot changed the title chore(deps): update dependency flax to v0.10.2 chore(deps): update dependency flax to v0.10.3 Feb 10, 2025
@renovate renovate bot force-pushed the renovate/flax-0.x-lockfile branch from 933ac1a to 8e15679 Compare February 27, 2025 01:45
@renovate renovate bot changed the title chore(deps): update dependency flax to v0.10.3 chore(deps): update dependency flax to v0.10.4 Feb 27, 2025
@renovate renovate bot force-pushed the renovate/flax-0.x-lockfile branch from 8e15679 to e8785ed Compare March 31, 2025 16:49
@renovate renovate bot changed the title chore(deps): update dependency flax to v0.10.4 chore(deps): update dependency flax to v0.10.5 Mar 31, 2025
@renovate renovate bot force-pushed the renovate/flax-0.x-lockfile branch from e8785ed to 821cd95 Compare April 2, 2025 18:22
@renovate renovate bot force-pushed the renovate/flax-0.x-lockfile branch from 821cd95 to 4aa34dd Compare April 23, 2025 23:57
@renovate renovate bot changed the title chore(deps): update dependency flax to v0.10.5 chore(deps): update dependency flax to v0.10.6 Apr 23, 2025
@renovate renovate bot force-pushed the renovate/flax-0.x-lockfile branch from 4aa34dd to f2abab1 Compare July 2, 2025 06:51
@renovate renovate bot changed the title chore(deps): update dependency flax to v0.10.6 chore(deps): update dependency flax to v0.10.7 Jul 2, 2025
@renovate renovate bot force-pushed the renovate/flax-0.x-lockfile branch from f2abab1 to e77fdbc Compare July 30, 2025 01:04
@renovate renovate bot changed the title chore(deps): update dependency flax to v0.10.7 chore(deps): update dependency flax to v0.11.0 Jul 30, 2025
@renovate renovate bot force-pushed the renovate/flax-0.x-lockfile branch from e77fdbc to d11b9b4 Compare August 9, 2025 01:06
@renovate renovate bot changed the title chore(deps): update dependency flax to v0.11.0 chore(deps): update dependency flax to v0.11.1 Aug 9, 2025
@renovate renovate bot force-pushed the renovate/flax-0.x-lockfile branch from d11b9b4 to 19f2316 Compare August 31, 2025 11:16
@renovate renovate bot changed the title chore(deps): update dependency flax to v0.11.1 chore(deps): update dependency flax to v0.11.2 Aug 31, 2025
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.

0 participants