Version 0.6.6

What's Changed

• Enhance checkpointing docs with Orbax information by @cpgaffney1 in #2879
• A small doc and one fixe to Imagenet benchmark by @nouiz in #2812
• Refactor and separate out name relaxation policy changes. by @levskaya in #2873
• Added RMS normalization layer by @chiamp in #2881
• added feature flag for FrozenDicts by @chiamp in #2872
• Explicitly mark the lazy-rng with legacy_config, and all new configs should use flax.config. by @copybara-service in #2895
• Added test for FrozenDict feature flag by @chiamp in #2896
• Added dict utility functions by @chiamp in #2897
• Improve notebook handling by @cgarciae in #2860
• migrate tests to regular dicts by @chiamp in #2898
• Release 0.6.6 by @IvyZX in #2908

New Contributors

• @cpgaffney1 made their first contribution in #2879

Full Changelog: v0.6.5...v0.6.6