0.12.11

What's Changed

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

New Contributors

• @jdymitarai made their first contribution in #5582
• @Anu27n made their first contribution in #5592
• @mayuriphad made their first contribution in #5593

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