diff --git a/docs/source/conf.py b/docs/source/conf.py
index 7a85d265..e4c97071 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -28,9 +28,9 @@
 author = 'Yahoo Inc'
 
 # The short X.Y version
-version = '2.0.0'
+version = '2.1.0'
 # The full version, including alpha/beta/rc tags
-release = '2.0.0'
+release = '2.1.0'
 
 
 # -- General configuration ---------------------------------------------------
diff --git a/pom.xml b/pom.xml
index 1b56730d..73594309 100644
--- a/pom.xml
+++ b/pom.xml
@@ -5,7 +5,7 @@
     <modelVersion>4.0.0</modelVersion>
     <groupId>com.yahoo.ml</groupId>
     <artifactId>tensorflowonspark</artifactId>
-    <version>2.0.0-SNAPSHOT</version>
+    <version>2.1.0-SNAPSHOT</version>
     <packaging>jar</packaging>
     <name>tensorflowonspark</name>
     <description>Spark Scala inferencing for TensorFlowOnSpark</description>
diff --git a/setup.py b/setup.py
index cc77ac0e..c11d32c3 100644
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@
 setup(
   name='tensorflowonspark',
   packages=['tensorflowonspark'],
-  version='2.0.0',
+  version='2.1.0',
   description='Deep learning with TensorFlow on Apache Spark clusters',
   long_description=long_description,
   long_description_content_type='text/markdown',
diff --git a/tensorflowonspark/__init__.py b/tensorflowonspark/__init__.py
index b655eb39..044335d1 100644
--- a/tensorflowonspark/__init__.py
+++ b/tensorflowonspark/__init__.py
@@ -2,4 +2,4 @@
 
 logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s (%(threadName)s-%(process)d) %(message)s")
 
-__version__ = "2.0.0"
+__version__ = "2.1.0"