diff --git a/docs/source/conf.py b/docs/source/conf.py
index f5982b82..5e0b468c 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -27,9 +27,9 @@
 author = 'Yahoo Inc'
 
 # The short X.Y version
-version = '1.3.4'
+version = '1.4.0'
 # The full version, including alpha/beta/rc tags
-release = '1.3.4'
+release = '1.4.0'
 
 
 # -- General configuration ---------------------------------------------------
diff --git a/setup.py b/setup.py
index d2fda920..2a04d70d 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'tensorflowonspark',
   packages = ['tensorflowonspark'],
-  version = '1.3.4',
+  version = '1.4.0',
   description = 'Deep learning with TensorFlow on Apache Spark clusters',
   author = 'Yahoo, Inc.',
   url = 'https://github.com/yahoo/TensorFlowOnSpark',
diff --git a/tensorflowonspark/__init__.py b/tensorflowonspark/__init__.py
index f29dc02c..33b5f8e6 100644
--- a/tensorflowonspark/__init__.py
+++ b/tensorflowonspark/__init__.py
@@ -1,3 +1,5 @@
 import logging
 
 logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s (%(threadName)s-%(process)d) %(message)s")
+
+__version__ = "1.4.0"