Update requirements.txt

ott-jax · Jan 31, 2022 · 74d519e · 74d519e
1 parent 76c8233
commit 74d519e
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -1,3 +1,10 @@
+absl-py>=0.7.0
+jax>=0.1.67
+jaxlib>=0.1.47
+numpy>=1.18.4
+matplotlib>=2.0.1
+flax>=0.3.6
+optax>=0.0.9
 matplotlib>=3.3.4
 pandoc>=1.0.2
 sphinx>=3.5.1