Merge pull request #2820 from IvyZX:push

PiperOrigin-RevId: 504950133

CHANGELOG.md

@@ -24,6 +24,22 @@ vNext
 -
 -
 
+0.6.4
+-----
+New features:
+- Our [ReadTheDoc](https://flax.readthedocs.io/en/latest/index.html) site is a lot more organized now! More improvements on the way.
+- Flax auto-SPMD parallelism API to work seamlessly with `jax.pjit`: https://flax.readthedocs.io/en/latest/guides/flax_on_pjit.html
+- Added new `zeros_init` and `ones_init` initializers.
+- Adds standardize initializer.
+- Allowed specifying method as a string.
+- Allowed runtime overwrite of `flax.config` flags.
+
+Bug fixes:
+- Added missing `dataclass.fields` from `__repr__`.
+- Renamed ConvLSTM to ConvLSTMCell.
+- Fix some tiny inconsistencies between scope.py and module.py.
+- Improved many many docstrings, comments and error messages.
+
 0.6.3
 -----
 New features:

README.md

@@ -191,8 +191,8 @@ 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.3},
-  year = {2020},
+  version = {0.6.4},
+  year = {2023},
 }
 ```
 

flax/version.py

@@ -13,5 +13,5 @@
 # limitations under the License.
 
 """Current Flax version at head on Github."""
-__version__ = "0.6.3"
+__version__ = "0.6.4"