diff --git a/pygmtools/__init__.py b/pygmtools/__init__.py
index b2b3534..0453380 100644
--- a/pygmtools/__init__.py
+++ b/pygmtools/__init__.py
@@ -15,7 +15,7 @@
 from .neural_solvers import pca_gm, ipca_gm, cie, ngm, genn_astar
 import pygmtools.utils as utils
 BACKEND = 'numpy'
-__version__ = '0.3.8'
+__version__ = '0.4.0'
 __author__ = 'ThinkLab at SJTU'