google/flax v0.6.5 on GitHub

What's Changed

Internal change by @copybara-service in #2826
Cache .mypy_cache on CI by @cgarciae in #2819
improve nn.map_variables docstrings by @cgarciae in #2818
Render non-array variables in tabulate by @cgarciae in #2769
Improve nn.map_variables example by @cgarciae in #2831
Introduce relaxed naming policy behind a feature flag. by @levskaya in #2833
Bumps minimal JAX version from 0.3.16 to 0.3.24 and cleans up some code. by @copybara-service in #2840
import initializers module directly by @chiamp in #2830
Re-box axis metadata in DenseGeneral's init functions by @IvyZX in #2843
Update jax dependency by @jheek in #2827
replaced zeros and ones initializers with zeros_init and ones_init builder functions by @chiamp in #2815
Fixing the punctuation in the README by @arcAman07 in #2811
Implement lazy init by @jheek in #2816
improve nn.checkpoint docs by @cgarciae in #2837
Flax 0.6.5 point release by @levskaya in #2856

Full Changelog: v0.6.4...v0.6.5