0.12.11

[cgarciae]

What's Changed

• Fix mypy hijax error by @samanklesaria in Fix mypy hijax error #5580
• Fix nnx.cond tracing cache misses by caching SimpleCondFn wrappers by @chenkuanliao in Fix nnx.cond tracing cache misses by caching SimpleCondFn wrappers #5518
• Add nnx_sow_capture_noop config flag and capture_noop option to Module.sow. by @copybara-service[bot] in Add nnx_sow_capture_noop config flag and capture_noop option to Module.sow. #5607
• Replace # pytype: disable suppressions with # pyrefly: ignore by @copybara-service[bot] in Replace # pytype: disable suppressions with # pyrefly: ignore #5612
• Fix Flax GitHub Actions CI (pytest DeprecationWarning and pytype in examples/gemma/train.py). by @copybara-service[bot] in Fix Flax GitHub Actions CI (pytest DeprecationWarning and pytype in examples/gemma/train.py). #5616
• Don't pass linen_meta_type argument when creating AxisMetadata subclasses. by @copybara-service[bot] in Don't pass linen_meta_type argument when creating AxisMetadata subclasses. #4933
• Add capture_noop option to Module.perturb. by @copybara-service[bot] in Add capture_noop option to Module.perturb. #5621
• Add sepreformer example by @samanklesaria in Add sepreformer example #5472
• Add multimodal llm example by @samanklesaria in Add multimodal llm example #5488
• Store nnx.WeightNorm scales as Param so they are trainable by @IvyZX in Store nnx.WeightNorm scales as Param so they are trainable #5623
• [nnx] Make nnx.Sequential iterable and filter **kwargs per layer. by @copybara-service[bot] in [nnx] Make nnx.Sequential iterable and filter **kwargs per layer. #5625
• Release Flax 0.12.11 by @copybara-service[bot] in Release Flax 0.12.11 #5627
• fix(nnx): skip Sequence modules in capture to prevent mixed key types (fixes nnx.capture fails with nnx.scan when the model contains nnx.List (mixed str/int pytree dict keys) #5571) by @jdymitarai in fix(nnx): skip Sequence modules in capture to prevent mixed key types (fixes #5571) #5582
• Fix NNX MultiHeadAttention is_causal mask during decode by @Anu27n in Fix NNX MultiHeadAttention is_causal mask during decode #5592
• fix(traverse_util): raise ValueError when flatten_dict separator conflicts with keys by @mayuriphad in fix(traverse_util): raise ValueError when flatten_dict separator conflicts with keys #5593

New Contributors

• @jdymitarai made their first contribution in fix(nnx): skip Sequence modules in capture to prevent mixed key types (fixes #5571) #5582
• @Anu27n made their first contribution in Fix NNX MultiHeadAttention is_causal mask during decode #5592
• @mayuriphad made their first contribution in fix(traverse_util): raise ValueError when flatten_dict separator conflicts with keys #5593

Full Changelog: v0.12.10...v0.12.11

---

This discussion was created from the release 0.12.11.