Merge pull request #3155 from IvyZX:push

PiperOrigin-RevId: 542060958
google · Jun 20, 2023 · fd567c8 · fd567c8
2 parents bdb68ef + 6ad44b7
commit fd567c8
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -22,6 +22,16 @@ vNext
 -
 -
 -
+0.6.11
+-----
+- Set Orbax-as-backend to be the default checkpointing method.
+- Fix setup trigger issue under sharing and transforms.
+- Add collection to self.scope.reserve(name, col) so that sow works with the same name in different collections.
+- Minor improvements for Sequential.
+- Improve the error message in MultiHeadDotProductAttention.
+- Allow manually specifying the rng key for Dropout.
+- RNN refactor.
+- fixed missing separator for rng fold in.
 
 0.6.10
 -----

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.6.9},
+ version = {0.6.11},
  year = {2023},
 }
 ```

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.10"
+__version__ = "0.6.11"