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/pyproject.toml
+++ b/pyproject.toml
@ -28,7 +28,10 @@ dependencies = [

 [project.optional-dependencies]
 cuda = [
-    "jax[cuda12]==0.9.0.1",
+    "jax[cuda13]==0.9.0.1",
+]
+analysis = [
+    "tensorboard",
 ]

 [dependency-groups]