diff --git a/rax/__init__.py b/rax/__init__.py
index 3f62c03..c711943 100644
--- a/rax/__init__.py
+++ b/rax/__init__.py
@@ -42,7 +42,7 @@
 from rax._src.t12n import gumbel_t12n
 from rax._src.t12n import segment_t12n
 
-__version__ = "0.3.0"
+__version__ = "0.4.0"
 
 __all__ = [
     "dcg2_lambdaweight",