diff --git a/jax_toolkit/_version.py b/jax_toolkit/_version.py
index 3dc1f76..485f44a 100644
--- a/jax_toolkit/_version.py
+++ b/jax_toolkit/_version.py
@@ -1 +1 @@
-__version__ = "0.1.0"
+__version__ = "0.1.1"
diff --git a/setup.cfg b/setup.cfg
index b82f934..b1c0d9e 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.1.0
+current_version = 0.1.1
 commit = True
 tag = True