diff --git a/setup.py b/setup.py
index 00b65e8..1ed341e 100644
--- a/setup.py
+++ b/setup.py
@@ -15,7 +15,7 @@
 
 setup(
     name="torchdyn",
-    version="0.4.3",
+    version="0.5.0",
     author="Michael Poli and Stefano Massaroli",
     description="PyTorch package for all things neural differential equations.",
     url="https://github.com/DiffEqML/torchdyn",