Merge pull request google#2134 from jheek:version-0.5.0

PiperOrigin-RevId: 450415939

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
 -----

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},
 }
 ```

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"