diff --git a/setup.py b/setup.py
index a68d4e61..e0f84e2a 100644
--- a/setup.py
+++ b/setup.py
@@ -39,13 +39,13 @@
 setup(
     name="torch_geometric_temporal",
     packages=find_packages(),
-    version="0.53.0",
+    version="0.54.0",
     license="MIT",
     description="A Temporal Extension Library for PyTorch Geometric.",
     author="Benedek Rozemberczki",
     author_email="benedek.rozemberczki@gmail.com",
     url="https://github.com/benedekrozemberczki/pytorch_geometric_temporal",
-    download_url="https://github.com/benedekrozemberczki/pytorch_geometric_temporal/archive/v0.53.0.tar.gz",
+    download_url="https://github.com/benedekrozemberczki/pytorch_geometric_temporal/archive/v0.54.0.tar.gz",
     keywords=keywords,
     install_requires=install_requires,
     extras_require={
diff --git a/torch_geometric_temporal/__init__.py b/torch_geometric_temporal/__init__.py
index a7fd2a9a..2e27d18e 100644
--- a/torch_geometric_temporal/__init__.py
+++ b/torch_geometric_temporal/__init__.py
@@ -2,7 +2,7 @@
 from torch_geometric_temporal.dataset import *
 from torch_geometric_temporal.signal import *
 
-__version__ = "0.53.0"
+__version__ = "0.54.0"
 
 __all__ = [
     "torch_geometric",