diff --git a/CHANGELOG.md b/CHANGELOG.md index 3e8ffc800..162f3a9dd 100644 --- 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 index b70e35c87..24fe49697 100644 --- 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}, } ```