Begin of 0.2.1 development

vroulet · OptaxDev · commit 54febd65982b · 2024-03-05T09:08:01.000-08:00
PiperOrigin-RevId: 612862356
diff --git a/optax/__init__.py b/optax/__init__.py
@@ -213,7 +213,7 @@
 )
 dpsgd = contrib.dpsgd
 
-__version__ = "0.2.0"
+__version__ = "0.2.1.dev"
 
 __all__ = (
     "adabelief",
