diff --git a/bnpm/__init__.py b/bnpm/__init__.py
index a994752..1b184ce 100644
--- a/bnpm/__init__.py
+++ b/bnpm/__init__.py
@@ -80,4 +80,4 @@ def prepare_cv2_imshow():
 for pkg in __all__:
     exec('from . import ' + pkg)
 
-__version__ = '0.5.3'
\ No newline at end of file
+__version__ = '0.5.4'
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index ab8cda2..1d76ef4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -26,6 +26,7 @@ sqlalchemy
 pymysql
 xxhash
 wandb
+tensorly
 torch
 torchvision
 torchaudio
diff --git a/setup.py b/setup.py
index 23eee1a..7e379a1 100644
--- a/setup.py
+++ b/setup.py
@@ -76,6 +76,7 @@ def read_requirements():
     'pytest',
     'hypothesis',
     'PyYAML',
+    'tensorly',
     'torch',
     'torchvision',
     'torchaudio',