diff --git a/optax/__init__.py b/optax/__init__.py index 41e2ff75..83d72c31 100644 --- a/optax/__init__.py +++ b/optax/__init__.py @@ -215,7 +215,7 @@ ) dpsgd = contrib.dpsgd -__version__ = "0.2.2.dev" +__version__ = "0.2.2" __all__ = ( "adabelief",