pypi / flax
flax
What changed in flax 0.12, from its official release notes, upgrade guides and advisories. Sources checked 2026-10-30.
1 breaking3 new APIs and options2 behavior and defaults5 fixes and other
See which of these apply to your code, ranked, in your agent:
npx @unstale/cli initflax 0.1211 changes
Breaking changes 1
Raises ValueError when the separator conflicts with keys.
New APIs and options 3
Adds a capture_noop option to Module.perturb.
Adds a capture_noop option to Module.sow.
Adds a config flag for controlling no-op capture in sow.
Behavior and default changes 2
Makes nnx.Sequential iterable and filters **kwargs per layer.
The HiPrim fallback now uses attribute lookup.
Fixes and other changes 5
Changes to traverse utilities were reverted, restoring the behavior before the breaking change.
Skips Sequence modules in capture to prevent mixed key types.
Fixes tracing cache misses by caching SimpleCondFn wrappers.
Fixes the is_causal mask during decode.
Stores WeightNorm scales as Param so they are trainable.