From c00a140812f7b7b27f23ca34e2a1247a0af08d98 Mon Sep 17 00:00:00 2001
From: Jonathan Heek <heekjonathan@gmail.com>
Date: Wed, 31 Mar 2021 09:23:30 +0000
Subject: [PATCH 1/3] Release 0.3.3

---
 CHANGELOG.md    | 27 ++++++++++++++++++++-------
 flax/version.py |  2 +-
 2 files changed, 21 insertions(+), 8 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 2ed4f0363..ac49cc1a8 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -7,8 +7,6 @@ vNext
 (Add your change to a random empty line to avoid merge conflicts)
  -
  -
- - Added custom error classes to many Linen errors. See: 
-   https://flax.readthedocs.io/en/latest/flax.errors.html
  -
  -
  -
@@ -16,19 +14,34 @@ vNext
  -
  -
  -
- - Bug Fix: Disallow modifying attributes in Modules after they are initialized.
- - Adds `Module.bind` for binding variables and RNGs to an interactive Module.
  -
  -
  -
- - Add option to overwrite existing checkpoints in `save_checkpoint`.
- - Raise an error when saving a checkpoint which has a smaller step than the
-   latest checkpoint already saved.
  -
  -
  -
  -
  -
+ -
+ -
+ -
+ -
+
+
+0.3.3
+------
+
+Possible breaking changes:
+ - Bug Fix: Disallow modifying attributes in Modules after they are initialized.
+ - Raise an error when saving a checkpoint which has a smaller step than the
+   latest checkpoint already saved.
+  
+Other changes:
+ - Added custom error classes to many Linen errors. See: 
+   https://flax.readthedocs.io/en/latest/flax.errors.html
+ - Adds `Module.bind` for binding variables and RNGs to an interactive Module.
+ - Add option to overwrite existing checkpoints in `save_checkpoint`.
+ - Remove JAX omnistaging check for forward compatibility.
 
 0.3.2
 ------
diff --git a/flax/version.py b/flax/version.py
index 284c66df8..34dcfec74 100644
--- a/flax/version.py
+++ b/flax/version.py
@@ -12,5 +12,5 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "0.3.2"
+__version__ = "0.3.3"
 

From e357948870b1eaf6f7a5754f186da46593cd6b3a Mon Sep 17 00:00:00 2001
From: Jonathan Heek <heekjonathan@gmail.com>
Date: Wed, 31 Mar 2021 11:24:30 +0000
Subject: [PATCH 2/3] Update BibTex

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 01323ffc9..ee3ebba28 100644
--- a/README.md
+++ b/README.md
@@ -153,7 +153,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.3.2},
+  version = {0.3.3},
   year = {2020},
 }
 ```

From df2f8a850100f154749eba921d24ac11ab6e4cd1 Mon Sep 17 00:00:00 2001
From: Jonathan Heek <heekjonathan@gmail.com>
Date: Wed, 31 Mar 2021 11:36:54 +0000
Subject: [PATCH 3/3] Add missing items in CHANGELOG

---
 CHANGELOG.md | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index ac49cc1a8..14c01d821 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -35,13 +35,18 @@ Possible breaking changes:
  - Bug Fix: Disallow modifying attributes in Modules after they are initialized.
  - Raise an error when saving a checkpoint which has a smaller step than the
    latest checkpoint already saved.
+ - MultiOptimizer now rejects the case where multiple sub optimizers update the
+   same parameter.
   
 Other changes:
  - Added custom error classes to many Linen errors. See: 
    https://flax.readthedocs.io/en/latest/flax.errors.html
  - Adds `Module.bind` for binding variables and RNGs to an interactive Module.
+ - Adds `nn.apply` and `nn.init` for transforming arbitrary functions that take a `linen.Module` as their first argument.
  - Add option to overwrite existing checkpoints in `save_checkpoint`.
  - Remove JAX omnistaging check for forward compatibility.
+ - Pathlib compatibility for checkpoint paths.
+ - `is_leaf` argument in `traverse_util.flatten_dict`
 
 0.3.2
 ------