Add flax and dm-haiku to docs requirements.txt.

e3nn · Nov 2, 2023 · 349254c · 349254c
1 parent 2375789
commit 349254c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -11,3 +11,5 @@ jupyter-sphinx
 matscipy
 jraph
 nequip_jax @ git+https://github.com/mariogeiger/[email protected]
+flax
+dm-haiku