From 3393e12edb53b975b8893f53dd2d75ddb33b3891 Mon Sep 17 00:00:00 2001
From: Anselm Levskaya <levskaya@google.com>
Date: Sat, 18 Jul 2020 07:34:43 -0700
Subject: [PATCH] mark new version and add notes to changelog for v0.2

---
 CHANGELOG.md | 15 +++++++++++----
 setup.py     |  2 +-
 2 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index b3e5c343c2..e478b2190c 100644
--- 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
index 759e6074ba..87fa8a3238 100644
--- 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: