diff --git a/README.md b/README.md
index d24e2ee8a..ed4058ce3 100644
--- a/README.md
+++ b/README.md
@@ -409,7 +409,7 @@ To cite this repository:
   author = {Tom Hennigan and Trevor Cai and Tamara Norman and Igor Babuschkin},
   title = {{H}aiku: {S}onnet for {JAX}},
   url = {http://github.com/deepmind/dm-haiku},
-  version = {0.0.1},
+  version = {0.0.2},
   year = {2020},
 }
 ```
diff --git a/haiku/__init__.py b/haiku/__init__.py
index 393088581..652b250ca 100644
--- a/haiku/__init__.py
+++ b/haiku/__init__.py
@@ -94,7 +94,7 @@
 from haiku._src.typing import Params
 from haiku._src.typing import State
 
-__version__ = "0.0.1"
+__version__ = "0.0.2"
 
 __all__ = (
     "AvgPool",