summaryrefslogtreecommitdiff
path: root/requirements.txt
diff options
context:
space:
mode:
Diffstat (limited to 'requirements.txt')
-rw-r--r--requirements.txt3
1 files changed, 3 insertions, 0 deletions
diff --git a/requirements.txt b/requirements.txt
index 331dd57..4353247 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,4 +7,7 @@ pymanopt==0.2.4; python_version <'3'
pymanopt; python_version >= '3'
cvxopt
scikit-learn
+torch
+jax
+jaxlib
pytest \ No newline at end of file