diff --git a/setup.py b/setup.py
index 3fbfd97..ac4c0d7 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='scalax',
-    version='0.1.2',
+    version='0.1.3',
     license='Apache-2.0',
     description='Scaling utilities for JAX.',
     url='https://github.com/young-geng/scalax',