feat(hpc): tensorboard analysis

2026-04-05 06:57:15 +02:00 · 2026-04-05 06:57:15 +02:00 · 75fd385a88
commit 75fd385a88
parent 059ac681ec
6 changed files with 180 additions and 4 deletions
--- a/env/hpc/requirements.txt
+++ b/env/hpc/requirements.txt
@ -5,7 +5,6 @@ flax>=0.12.2
 gymnasium>=1.2.3
 ipykernel==7.2.0
 jax[cuda13]==0.9.0.1
-torch
 numpy>=2.0.0
 protobuf>=5.0.0
 warp-lang
@ -17,3 +16,4 @@ pyopengl>=3.1.10
 pyopengl-accelerate>=3.1.10
 tyro>=1.0.10
 wandb==0.24.2
+torch>=2.4.0