diff --git a/src/powerpax/__init__.py b/src/powerpax/__init__.py
index a66b939..040db05 100644
--- a/src/powerpax/__init__.py
+++ b/src/powerpax/__init__.py
@@ -10,7 +10,7 @@
 iterations (:func:`ppx.sliced_scan <powerpax.sliced_scan>`).
 """
 
-__version__ = "0.2.0.dev"
+__version__ = "0.1.1"
 __all__ = ["chunked_vmap", "sliced_scan", "checkpoint_chunked_scan", "Static"]
 
 from ._vmap import chunked_vmap