fix(hpc): modules loading

2026-04-04 21:11:53 +02:00 · 2026-04-04 21:11:53 +02:00 · 7d81f94835
commit 7d81f94835
parent 745c09e524
4 changed files with 38 additions and 38 deletions
--- a/env/hpc/requirements.txt
+++ b/env/hpc/requirements.txt
@ -1,9 +1,14 @@
 biorobot==0.4.2
 cleanrl>=0.4.8
 evosax==0.2.0
+flax>=0.12.2
 gymnasium>=1.2.3
 ipykernel==7.2.0
+jax==0.9.0.1
+matplotlib==3.10.8
 mediapy==1.2.6
+optax>=0.2.6
 pyopengl>=3.1.10
 pyopengl-accelerate>=3.1.10
 tyro>=1.0.10
+wandb==0.24.2