diff --git a/pyproject.toml b/pyproject.toml
index 5c62fb1..8848e71 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pynajax"
-version = "0.1.3"
+version = "0.2.0"
 authors = [{name = "pynajax authors"}]
 description = "A JAX backend for pynapple."
 readme = "README.md"
diff --git a/src/pynajax/__init__.py b/src/pynajax/__init__.py
index 4f13784..80d2f21 100644
--- a/src/pynajax/__init__.py
+++ b/src/pynajax/__init__.py
@@ -1,3 +1,3 @@
-__version__ = "0.1.3"
+__version__ = "0.2.0"
 # from .jax_core_bin_average import bin_average
 # from .jax_core_convolve import convolve