diff --git a/optax/__init__.py b/optax/__init__.py
index 322871633..a1f440c0e 100644
--- a/optax/__init__.py
+++ b/optax/__init__.py
@@ -144,7 +144,7 @@
 from optax._src.wrappers import MultiSteps
 from optax._src.wrappers import MultiStepsState
 
-__version__ = "0.0.8"
+__version__ = "0.0.9"
 
 __all__ = (
     "adabelief",