diff --git a/optax/__init__.py b/optax/__init__.py
index 3b42498bf..79cece717 100644
--- a/optax/__init__.py
+++ b/optax/__init__.py
@@ -204,7 +204,7 @@
 )
 dpsgd = contrib.dpsgd
 
-__version__ = "0.1.9"
+__version__ = "0.2.0.dev"
 
 __all__ = (
     "adabelief",