Download Latest Version
0.12.0 source code.tar.gz (5.0 MB)
Get an email when there's a new version of Flax
| Name | Modified | Size | Downloads / Week |
|---|---|---|---|
| Parent folder | |||
| README.md | 2025-08-08 | 1.4 kB | |
| v0.11.1 source code.tar.gz | 2025-08-08 | 5.2 MB | |
| v0.11.1 source code.zip | 2025-08-08 | 5.5 MB | |
| Totals: 3 Items | 10.7 MB | 0 | |
What's Changed
- Make
Sequential()be identity by @SobhanMP in https://github.com/google/flax/pull/4796 - Add a JAX/Flax key concepts doc by @IvyZX in https://github.com/google/flax/pull/4795
- miscellaneous improvements by @cgarciae in https://github.com/google/flax/pull/4859
- Replace
jax.sharding.use_meshwithjax.set_mesh.jax.set_meshcan act as a global setter or a context manager. by @copybara-service[bot] in https://github.com/google/flax/pull/4862 - Pytree and ArrayRef refactor by @cgarciae in https://github.com/google/flax/pull/4863
- Add old property attributes for object->pytree rename. by @copybara-service[bot] in https://github.com/google/flax/pull/4864
- Add BatchNorm layers to CNN in MNIST tutorial for improved training stability by @sanepunk in https://github.com/google/flax/pull/4773
- Description by @copybara-service[bot] in https://github.com/google/flax/pull/4866
- update and pop for dict by @cgarciae in https://github.com/google/flax/pull/4869
- simplify nnx_basics by @cgarciae in https://github.com/google/flax/pull/4868
- updates to version 0.11.1 by @cgarciae in https://github.com/google/flax/pull/4878
New Contributors
- @SobhanMP made their first contribution in https://github.com/google/flax/pull/4796
- @sanepunk made their first contribution in https://github.com/google/flax/pull/4773
Full Changelog: https://github.com/google/flax/compare/v0.11.0...v0.11.1