Merge pull request #363 from levskaya:v0.2

PiperOrigin-RevId: 321967880
google · Jul 18, 2020 · 79507ec · 79507ec
2 parents a5dfa29 + 3393e12
commit 79507ec
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,14 @@
-vNEXT
----
+Changelog
+----------
+
+
+v0.2
+----
+ - Added JAX trace-level checks for transforms.
+ - BatchNorm added axis_index_groups for control in parallel training.
+ - Optimizers broken out into separate directory with base class and implementations.
+ - traverse_util added flatten_dict and unflatten_dict utility methods for nested dicts.
+
 
 v0.1
 ----
@@ -12,5 +21,3 @@ v0.1
  - Change default layer names from numbers '0', '1', etc. to
    include the Module class name, e.g. 'Dense_0', 'LayerNorm_1'.
 
-### More
-...
diff --git a/setup.py b/setup.py
@@ -18,7 +18,7 @@
 from setuptools import find_packages
 from setuptools import setup
 
-version = "0.1.0"
+version = "0.2.0"
 
 here = os.path.abspath(os.path.dirname(__file__))
 try: