From 88934902dc4e400bc57f49168a43dc7709e74478 Mon Sep 17 00:00:00 2001
From: Marcus Chiam <marcuschiam@google.com>
Date: Fri, 28 Jul 2023 17:25:37 -0700
Subject: [PATCH] 0.7.1 release

---
 CHANGELOG.md    | 29 +++++++++++++++++++++++++++++
 README.md       |  2 +-
 flax/version.py |  2 +-
 3 files changed, 31 insertions(+), 2 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 93673f9b6b..09fbf3a074 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -23,6 +23,35 @@ vNext
 -
 -
 
+0.7.1
+-----
+Breaking changes:
+- Migrating Flax from returning FrozenDicts to returning regular dicts. More details can be found in this [announcement](https://github.com/google/flax/discussions/3191)
+
+New features:
+- Use pyink
+- added dict migration guide to index
+- add scan over layers section
+- Expose options to customize rich.Table
+- add support for initializing carry variables in scan
+- Let Flax-Orbax to not port the shape of `target` arrays when they port the `target` shardings.
+
+Bug fixes:
+- Use import `orbax.checkpoint` which is a better import pattern.
+- Use import `orbax.checkpoint as ocp` to avoid the verbosity of using 'orbax.checkpoint` every time.
+- [linen] Add alternative, more numerically stable, variance calculation to `LayerNorm`.
+- [linen] Minor cleanup to normalization code.
+- Fix norm calculation bug for 0-rank arrays.
+- [JAX] Remove references to jax.config.jax_array.
+- [linen] Use `stack` instead of `concatenate` in `compute_stats`, to handle scalar stats case.
+- [linen] More minor cleanup in normalization `compute_stats`.
+- Fix warnings from atari gym.
+- Refactor TypeHandler to operate over batches of values, rather than individual ones. This allows more flexibility for implementations that may operate more efficiently on batches.
+- Fix carry slice logic
+- make flax_basics guide use utility fns
+- Fix checkpointing guide error at head
+- Improve scan docs
+
 0.7.0
 -----
 - RNNCellBase refactor.
diff --git a/README.md b/README.md
index adf8902cb0..ac7b9433ba 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.0},
+  version = {0.7.1},
   year = {2023},
 }
 ```
diff --git a/flax/version.py b/flax/version.py
index 9c5e6150f2..0fb395cb9a 100644
--- a/flax/version.py
+++ b/flax/version.py
@@ -13,4 +13,4 @@
 # limitations under the License.
 
 """Current Flax version at head on Github."""
-__version__ = "0.7.0"
+__version__ = "0.7.1"