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