diff --git a/unit_scaling/_version.py b/unit_scaling/_version.py
index 783db5e..705bfd6 100644
--- a/unit_scaling/_version.py
+++ b/unit_scaling/_version.py
@@ -1,3 +1,3 @@
 # Copyright (c) 2024 Graphcore Ltd. All rights reserved.
 
-__version__ = "0.1"
+__version__ = "0.2"