Merge pull request #2705 from IvyZX:push

PiperOrigin-RevId: 494237571
google · Dec 9, 2022 · 2946b3c · 2946b3c
2 parents fec10eb + 2ea5388
commit 2946b3c
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -24,6 +24,18 @@ vNext
 -
 -
 
+0.6.3
+-----
+New features:
+- Flax checkpointing now uses [Orbax](https://github.com/google/orbax) for more flexiblity and features.
+- Added support for python 3.10 and removed support for 3.7.
+
+Bug fixes:
+- Fixed rng generation in DenseGeneral init.
+- Improved support for Mac M1 chip.
+- Bumped package versions for a bunch of examples.
+- Improved many docstrings and error messages.
+
 0.6.2
 -----
 New features:

diff --git a/README.md b/README.md
@@ -191,7 +191,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.2},
+  version = {0.6.3},
   year = {2020},
 }
 ```

diff --git a/flax/version.py b/flax/version.py
@@ -13,5 +13,5 @@
 # limitations under the License.
 
 """Current Flax version at head on Github."""
-__version__ = "0.6.2"
+__version__ = "0.6.3"