Merge pull request #2944 from IvyZX:push

PiperOrigin-RevId: 515489629

CHANGELOG.md

@@ -24,6 +24,15 @@ vNext
 -
 -
 
+0.6.7
+-----
+- New checkpoints will be saved using Orbax! Please check out [upgrade guide](https://flax.readthedocs.io/en/latest/guides/orbax_upgrade_guide.html) and consider migrating completely to the Orbax API.
+  - You could `flax.config.update('flax_use_orbax_checkpointing', False)` to temporarily disable this migration, but note that Flax legacy checkpointing will be removed 3 months from Mar 10, 2023.
+- Migrating `FrozenDict` to regular dict: utility functions now work on both.
+- Migrated Flax dataclass and `FrozenDict` to JAX pytree keypath API.
+- Fixed pytype and improved typing for `Module`
+- Fixed up uses of PyTree and PyTreeDef types.
+
 0.6.6
 -----
 - 0.6.5 was yanked so this release contains all that was in 0.6.5 as well.

README.md

@@ -197,7 +197,7 @@ To cite this repository:
   author = {Jonathan Heek and Anselm Levskaya and Avital Oliver and Marvin Ritter and Bertrand Rondepierre and Andreas Steiner and Marc van {Z}ee},
   title = {{F}lax: A neural network library and ecosystem for {JAX}},
   url = {http://github.com/google/flax},
-  version = {0.6.6},
+  version = {0.6.7},
   year = {2023},
 }
 ```