diff --git a/dm_pix/__init__.py b/dm_pix/__init__.py
index e750c65..2168fa2 100644
--- a/dm_pix/__init__.py
+++ b/dm_pix/__init__.py
@@ -20,7 +20,7 @@
 from dm_pix._src import metrics
 from dm_pix._src import patch
 
-__version__ = "0.3.4"
+__version__ = "0.4.0"
 
 # Augmentations.
 adjust_brightness = augment.adjust_brightness