diff --git a/.bumpversion.cfg b/.bumpversion.cfg
index b588542..0910806 100644
--- a/.bumpversion.cfg
+++ b/.bumpversion.cfg
@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.3.1
+current_version = 0.4.0
 commit = True
 tag = True
 parse = (?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)
diff --git a/docs/source/conf.py b/docs/source/conf.py
index 27c7c4a..87d8370 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -23,7 +23,7 @@ author = "Jensun Ravichandran"
 
 # The full version, including alpha/beta/rc tags
 #
-release = "0.3.1"
+release = "0.4.0"
 
 # -- General configuration ---------------------------------------------------
 
diff --git a/prototorch/__init__.py b/prototorch/__init__.py
index 19312a4..a7c4866 100644
--- a/prototorch/__init__.py
+++ b/prototorch/__init__.py
@@ -1,7 +1,7 @@
 """ProtoTorch package."""
 
 # Core Setup
-__version__ = "0.3.1"
+__version__ = "0.4.0"
 
 __all_core__ = [
     "datasets",
diff --git a/setup.py b/setup.py
index 5fff30c..efdddff 100644
--- a/setup.py
+++ b/setup.py
@@ -41,7 +41,7 @@ ALL = DOCS + DATASETS + EXAMPLES + TESTS
 
 setup(
     name="prototorch",
-    version="0.3.1",
+    version="0.4.0",
     description="Highly extensible, GPU-supported "
     "Learning Vector Quantization (LVQ) toolbox "
     "built using PyTorch and its nn API.",