From 1275c716f10112a60edc0c8487ecebf22880c280 Mon Sep 17 00:00:00 2001
From: jheek <heekjonathan@gmail.com>
Date: Mon, 23 May 2022 10:25:40 +0200
Subject: [PATCH] Version 0.5.0

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

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 961806cb2..0bac207f2 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -12,13 +12,8 @@ vNext
 -
 -
 -
-- Implemented [default dtype FLIP](https://github.com/google/flax/blob/main/docs/flip/1777-default-dtype.md).
-  This means the default dtype is now inferred from inputs and params rather than being hard-coded to float32.
-  This is especially useful for dealing with complex numbers because the standard Modules will no longer truncate
-  complex numbers to their real component by default. Instead the complex dtype is preserved by default.
 -
 -
-- Added `flax.jax_utils.ad_shard_unpad()` by @lucasb-eyer
 -
 -
 -
@@ -28,6 +23,23 @@ vNext
 -
 -
 -
+-
+-
+
+0.5.0
+-----
+- Added `flax.jax_utils.ad_shard_unpad()` by @lucasb-eyer
+- Implemented [default dtype FLIP](https://github.com/google/flax/blob/main/docs/flip/1777-default-dtype.md).
+  This means the default dtype is now inferred from inputs and params rather than being hard-coded to float32.
+  This is especially useful for dealing with complex numbers because the standard Modules will no longer truncate
+  complex numbers to their real component by default. Instead the complex dtype is preserved by default.
+
+
+Bug fixes:
+- Fix support for JAX's experimental_name_stack.
+
+Breaking changes:
+- In rare cases the dtype of a layer can change due to  [default dtype FLIP](https://github.com/google/flax/blob/main/docs/flip/1777-default-dtype.md). See the "Backward compatibility" section of the proposal for more information.
 
 0.4.2
 -----
diff --git a/README.md b/README.md
index 649c77fdd..defc472b2 100644
--- a/README.md
+++ b/README.md
@@ -190,7 +190,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.2},
+  version = {0.5.0},
   year = {2020},
 }
 ```
diff --git a/flax/version.py b/flax/version.py
index af09a08bb..06e8f3f98 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.2"
+__version__ = "0.5.0"