From 0377fd4ec1bc7578db0b1942f111f12474a325dc Mon Sep 17 00:00:00 2001
From: Andreas Steiner <andstein@google.com>
Date: Wed, 23 Mar 2022 11:04:13 +0100
Subject: [PATCH] Bumps version to 0.4.1

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

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 933ee5a0..a442a788 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -9,13 +9,24 @@ vNext
 -
 -
 -
+-
+-
+-
+-
+-
+-
+-
+-
+
+0.4.1
+-----
+
+New features:
+- Added locally-connected (unshared CNN) layer `flax.linen.ConvLocal`.
 - Improved seq2seq example: Factored our model and input pipeline code.
 - Added Optax update guide and deprecated `flax.optim`.
 - Added `sep` argument to `flax.traverse_util.flatten_dict()`.
 - Implemented Sequential module, in `flax.linen.combinators`.
--
-- Added locally-connected (unshared CNN) layer `flax.linen.ConvLocal`.
--
 
 0.4.0
 ------
diff --git a/README.md b/README.md
index e1a561fa..cf9c9f8a 100644
--- a/README.md
+++ b/README.md
@@ -186,7 +186,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.4.0},
+  version = {0.4.1},
   year = {2020},
 }
 ```
diff --git a/flax/version.py b/flax/version.py
index 7078ae48..6c5e36c2 100644
--- 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.4.0"
+__version__ = "0.4.1"