diff --git a/setup.py b/setup.py
index 3b1c86e..d1efe16 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='pytorchts',
-    version='0.1.1',
+    version='0.2.0',
     description="PyTorch Probabilistic Time Series Modeling framework",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",