diff --git a/docs/source/conf.py b/docs/source/conf.py
index 0c822222..45f94b13 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -15,7 +15,7 @@
 project = "Apax"
 copyright = "2023, Moritz Schäfer, Nico Segreto, Johannes Kästner"
 author = "Moritz Schäfer, Nico Segreto, Johannes Kästner"
-release = "0.1.0"
+release = "0.4.0"
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration
diff --git a/pyproject.toml b/pyproject.toml
index 8fae38a3..7043c071 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "apax"
-version = "0.3.0"
+version = "0.4.0"
 description = "Atomistic Learned Potential Package in JAX"
 authors = ["Moritz René Schäfer <schaefer@theochem.uni-stuttgart.de>", "Nico Segreto <segreto@theochem.uni-stuttgart.de>"]
 keywords=["machine-learning", "interatomic potentials", "molecular-dynamics"]