diff --git a/pyproject.toml b/pyproject.toml
index cb3b673..06cc469 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "torchdyn"
-version = "1.0.5"
+version = "1.0.6"
 license = "Apache License, Version 2.0"
 description = "A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods."
 authors = ["Michael Poli", "Stefano Massaroli", "DiffEqML"]
diff --git a/setup.py b/setup.py
index 0cf51e7..0b89305 100644
--- a/setup.py
+++ b/setup.py
@@ -15,7 +15,7 @@
 
 setup(
     name="torchdyn",
-    version="1.0.5",
+    version="1.0.6",
     author="Michael Poli and Stefano Massaroli",
     description="PyTorch package for all things neural differential equations.",
     url="https://github.com/DiffEqML/torchdyn",
diff --git a/torchdyn/__init__.py b/torchdyn/__init__.py
index 3e1c2e3..47bda20 100644
--- a/torchdyn/__init__.py
+++ b/torchdyn/__init__.py
@@ -10,7 +10,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = '1.0.5'
+__version__ = '1.0.6'
 __author__  = 'Michael Poli, Stefano Massaroli et al.'
 
 from torch import Tensor