add jax

condecon · Jan 4, 2025 · 30ac520 · 30ac520
1 parent 57897bc
commit 30ac520
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,4 @@
+jax
 numpy>=1.20.0
 scipy>=1.13.0
 pandas>=2.2.0

diff --git a/setup.py b/setup.py
@@ -25,6 +25,7 @@
               "adaptivetesting.simulation",
               "adaptivetesting.tests"],
     install_requires=[
+        "jax",
         "numpy>=1.20.0",
         "scipy>=1.13.0",
         "pandas>=2.2.0",