diff --git a/optax/__init__.py b/optax/__init__.py index 18e0fac48..0b3f590ff 100644 --- a/optax/__init__.py +++ b/optax/__init__.py @@ -178,7 +178,7 @@ from optax._src.wrappers import skip_large_updates from optax._src.wrappers import skip_not_finite -__version__ = "0.1.3" +__version__ = "0.1.4" __all__ = ( "adabelief",