diff --git a/optax/__init__.py b/optax/__init__.py
index a94627461..1622b0f36 100644
--- a/optax/__init__.py
+++ b/optax/__init__.py
@@ -213,7 +213,7 @@
 )
 dpsgd = contrib.dpsgd
 
-__version__ = "0.2.1.dev"
+__version__ = "0.2.1"
 
 __all__ = (
     "adabelief",