requirements

gerkone · Jul 27, 2023 · ca691c1 · ca691c1
1 parent 9128b27
commit ca691c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,6 @@
 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
-dm-haiku==0.0.9
-e3nn-jax==0.19.3
+dm-haiku>=0.0.9
+e3nn-jax>=0.17.4
 jax[cuda]
 jraph==0.0.6.dev0
 numpy>=1.23.4