diff --git a/setup.py b/setup.py
index 7ce1185..2497952 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='nanodl',
-    version='1.0.0.dev1',
+    version='1.0.1.dev1',
     author='Henry Ndubuaku',
     author_email='ndubuakuhenry@gmail.com',
     description='A Jax-based library for designing and training transformer models from scratch.',