Merge pull request #2103 from marcvanzee:v0_4_2

PiperOrigin-RevId: 446682006

CHANGELOG.md

@@ -8,15 +8,42 @@ vNext
 -
 -
 -
-- Added lifted conditional `nn.cond`
-- Handle rate==1.0 edgecase in Dropout.
 -
 -
 -
 -
 -
 -
 -
+-
+-
+-
+-
+-
+-
+-
+-
+-
+-
+
+0.4.2
+-----
+
+New features:
+- Add lifted conditional `nn.cond`.
+- Improved error messages: parameters not found, loading checkpoints.
+- Replace `jax.tree_multimap` (deprecated) with `jax.tree_map`.
+- Add the "Module Lifecycle" design note.
+- Add support for JAX dynamic stack-based named_call
+
+Bug fixes:
+- Handle rate==1.0 edgecase in Dropout.
+- Fix bug where Linen Module state is reused.
+- Bug fixes and generalizations of nn.partitioning API.
+
+Breaking changes:
+- 
+
 
 0.4.1
 -----

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.1},
+  version = {0.4.2},
   year = {2020},
 }
 ```

flax/version.py

@@ -13,5 +13,5 @@
 # limitations under the License.
 
 """Current Flax version at head on Github."""
-__version__ = "0.4.1"
+__version__ = "0.4.2"