fix(hpc): rework modules again

2026-04-04 23:45:57 +02:00 · 2026-04-04 23:45:57 +02:00 · a705154617
commit a705154617
parent c8b4df71bb
3 changed files with 9 additions and 6 deletions
--- a/env/hpc/requirements.txt
+++ b/env/hpc/requirements.txt
@ -4,7 +4,8 @@ evosax==0.2.0
 flax>=0.12.2
 gymnasium>=1.2.3
 ipykernel==7.2.0
-jax==0.9.0.1
+jax[cuda13]==0.9.0.1
+scipy>=1.14.0
 numpy>=2.0.0
 protobuf>=5.0.0
 warp-lang