diff --git a/torchax/__init__.py b/torchax/__init__.py
index 481c4e3..af915f3 100644
--- a/torchax/__init__.py
+++ b/torchax/__init__.py
@@ -29,7 +29,7 @@
 
 from .checkpoint import load_checkpoint, save_checkpoint
 
-__version__ = "0.0.12"
+__version__ = "0.0.13"
 VERSION = __version__
 
 # the "fast path" uses some sparse tensor thingies that currently we