diff --git a/setup.py b/setup.py
index 1e89eb0..aebfbdc 100644
--- a/setup.py
+++ b/setup.py
@@ -25,7 +25,10 @@
         'sympy',
         'pyyaml',
         'opt_einsum',
-        'scikit-learn'
+        'scikit-learn',
+        'rich',
+        'tensorboard',
+        'tensorboardx'
     ],
     classifiers=[
         'Development Status :: 5 - Production/Stable',