diff --git a/pyproject.toml b/pyproject.toml
index 551b7bc..1367724 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -48,6 +48,7 @@ dependencies = [
     'torch',
     'matplotlib',
     'pytorch_seed',
+    'arm_pytorch_utilities',
 ]
 
 [project.optional-dependencies]