Release Version 0.7.1 · google/flax

What's Changed

added dict migration guide to index by @chiamp in #3188
[linen] Add alternative, more numerically stable, variance calculation to LayerNorm. by @copybara-service in #3194
[linen] Minor cleanup to normalization code. by @copybara-service in #3200
Fix warnings from atari gym. by @levskaya in #3207
Fix carry slice logic by @cgarciae in #3213
removed FrozenDict section by @chiamp in #3210
make flax_basics guide use utility fns by @chiamp in #3214
Fix typo in struct.py documentation by @marcoselvi in #3209
updated flax guides to use utility fns by @chiamp in #3215
[linen] More minor cleanup in normalization compute_stats. by @copybara-service in #3205
Expose options to customize rich.Table by @cgarciae in #3197
Use pyink by @cgarciae in #3216
Fix remaining pyink issues by @cgarciae in #3219
add scan over layers section by @cgarciae in #3195
Fix checkpointing guide error at head by @IvyZX in #3223
Improve scan docs by @cgarciae in #3231
0.7.1 release by @chiamp in #3234

Full Changelog: v0.7.0...v0.7.1