What's Changed
- Fix lm1b sampler by @cgarciae in #2842
- RNN Module by @cgarciae in #2604
- Add Orbax migration guide by @IvyZX in #2885
- Add 'all' install group by @cgarciae in #2894
- Improve nn.scan docs by @cgarciae in #2839
- make utility functions work on FrozenDicts and regular dicts by @chiamp in #2919
- Improve Flax dark mode - remove hard-coded text color by @8bitmp3 in #2924
- Fix pytype and improve typing for Module by @cgarciae in #2933
- Make flax.version 1 + the current release by @IvyZX in #2936
- Migrate Flax dataclass to the newest JAX pytree keypath API. by @copybara-service in #2921
- release 0.6.7 by @IvyZX in #2944
Full Changelog: v0.6.6...v0.6.7