Merge pull request #3326 from IvyZX:push

PiperOrigin-RevId: 564889489
google · Sep 13, 2023 · 5f06f08 · 5f06f08
2 parents 9aeb909 + ad8d22e
commit 5f06f08
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -23,6 +23,21 @@ vNext
 -
 -
 
+0.7.3
+-----
+New features:
+- Add QK-normalization to MultiHeadDotProductAttention
+- Allow apply's method argument to accept submodules
+- Add module path to nn.module.
+- [JAX] Generate new type of PRNG keys
+
+Bug fixes:
+- Directly call original method if method interceptor stack is empty.
+- fix stackoverflow when loading pickled module
+- Improve kw_only_dataclass.
+- Allow pass-through implementation of state dict
+- Promote dot_general injections from a function to a module.
+
 0.7.2
 -----
 New features:

diff --git a/README.md b/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.7.2},
+  version = {0.7.3},
   year = {2023},
 }
 ```