diff --git a/declearn/__init__.py b/declearn/__init__.py
index edb678d4c14353aaa9566041048209f2d5f12859..95315fd688c0e171702fbba726372873a84f4b7a 100644
--- a/declearn/__init__.py
+++ b/declearn/__init__.py
@@ -65,4 +65,4 @@ from . import (
     utils,
 )
 
-__version__ = "2.1.0"
+__version__ = "2.2.0"
diff --git a/pyproject.toml b/pyproject.toml
index 3ce5cc8b5391c0843c2df7bae7ff52d6c913193d..898e684035bccd908f481c7c094f7b7f4f553d2b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -8,7 +8,7 @@ requires = [
 
 [project]
 name = "declearn"
-version = "2.1.0"
+version = "2.2.0"
 description = "Declearn - a python package for private decentralized learning."
 readme = "README.md"
 requires-python = ">=3.8"