diff --git a/setup.py b/setup.py
index 6e2128b..fa2e35e 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="pytorchts",
-    version="0.3.1",
+    version="0.4.0",
     description="PyTorch Probabilistic Time Series Modeling framework",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",