diff --git a/flax/version.py b/flax/version.py
index e8a33d8413..f3e42a03c1 100644
--- a/flax/version.py
+++ b/flax/version.py
@@ -13,4 +13,4 @@
 # limitations under the License.
 
 """Current Flax version at head on Github."""
-__version__ = '0.8.1'
+__version__ = '0.8.2'