What's Changed
- [flax:examples:wmt] Small linter fixes. by @copybara-service[bot] in #5012
- [flax:examples:seq2seq] Create main and default config based on seq2seq.ipynb. by @copybara-service[bot] in #5119
- [flax:examples:vae] Small linter fixes. by @copybara-service[bot] in #5014
- [flax:examples:gemma] Fixing linter errors. by @copybara-service[bot] in #5013
- [flax:examples:sst2] Fix pytype errors. by @copybara-service[bot] in #5118
- Allow substring matching in
nnx.PathContainsby @thijs-vanweezel in #5094 - [flax:examples:sst2] Fix notebook error. by @copybara-service[bot] in #5122
- [flax:examples:ppo] Fix some linter / import issues. #jax-fixit by @copybara-service[bot] in #5120
- Avoid passing
concreteargument tojax.rematby @copybara-service[bot] in #5121 - [flax:examples:lm1b_nnx] Update example to work internally. #jax-fixit. by @copybara-service[bot] in #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 #5126
- [jax:benchmarks] Add tracing/lowering benchmarks for a few flax examples. by @copybara-service[bot] in #4911
- remove abstracted_axes from nnx.jit by @copybara-service[bot] in #5132
- Pooling operation by @jorisSchaller in #5057
- Added is_causal mask argument to flax.nnx.dot_product_attention by @ibbyml in #5093
- Add out_sharding argument to call methods for layers with jax calls that support it by @samanklesaria in #5102
- Temporary fix for failing CI by @vfdev-5 in #5144
- New release 0.12.2 by @IvyZX in #5149
New Contributors
- @thijs-vanweezel made their first contribution in #5094
- @ibbyml made their first contribution in #5093
Full Changelog: v0.12.1...v0.12.2