Version 0.12.2

[IvyZX]

What's Changed

• [flax:examples:wmt] Small linter fixes. by @copybara-service[bot] in [flax:examples:wmt] Small linter fixes. #5012
• [flax:examples:seq2seq] Create main and default config based on seq2seq.ipynb. by @copybara-service[bot] in [flax:examples:seq2seq] Create main and default config based on seq2seq.ipynb. #5119
• [flax:examples:vae] Small linter fixes. by @copybara-service[bot] in [flax:examples:vae] Small linter fixes. #5014
• [flax:examples:gemma] Fixing linter errors. by @copybara-service[bot] in [flax:examples:gemma] Fixing linter errors. #5013
• [flax:examples:sst2] Fix pytype errors. by @copybara-service[bot] in [flax:examples:sst2] Fix pytype errors. #5118
• Allow substring matching in nnx.PathContains by @thijs-vanweezel in Allow substring matching in nnx.PathContains #5094
• [flax:examples:sst2] Fix notebook error. by @copybara-service[bot] in [flax:examples:sst2] Fix notebook error. #5122
• [flax:examples:ppo] Fix some linter / import issues. #jax-fixit by @copybara-service[bot] in [flax:examples:ppo] Fix some linter / import issues. #jax-fixit #5120
• Avoid passing concrete argument to jax.remat by @copybara-service[bot] in Avoid passing concrete argument to jax.remat #5121
• [flax:examples:lm1b_nnx] Update example to work internally. #jax-fixit. by @copybara-service[bot] in [flax:examples:lm1b_nnx] Update example to work internally. #jax-fixit. #5125
• [flax:examples:nlp_seq] Create a main.py file to run tests with config files to match other examples. #jax-fixit by @copybara-service[bot] in [flax:examples:nlp_seq] Create a main.py file to run tests with config files to match other examples. #jax-fixit #5126
• [jax:benchmarks] Add tracing/lowering benchmarks for a few flax examples. by @copybara-service[bot] in [jax:benchmarks] Add tracing/lowering benchmarks for a few flax examples. #4911
• remove abstracted_axes from nnx.jit by @copybara-service[bot] in remove abstracted_axes from nnx.jit #5132
• Pooling operation by @jorisSchaller in Pooling operation  #5057
• Added is_causal mask argument to flax.nnx.dot_product_attention by @ibbyml in Added is_causal mask argument to flax.nnx.dot_product_attention #5093
• Add out_sharding argument to call methods for layers with jax calls that support it by @samanklesaria in Add out_sharding argument to call methods for layers with jax calls that support it #5102
• Temporary fix for failing CI by @vfdev-5 in Temporary fix for failing CI #5144
• New release 0.12.2 by @IvyZX in New release 0.12.2 #5149

New Contributors

• @thijs-vanweezel made their first contribution in Allow substring matching in nnx.PathContains #5094
• @ibbyml made their first contribution in Added is_causal mask argument to flax.nnx.dot_product_attention #5093

Full Changelog: v0.12.1...v0.12.2

---

This discussion was created from the release Version 0.12.2.