feat: Time+memory tracking

2025-12-07 21:49:45 +01:00 · 2025-12-07 21:49:45 +01:00 · ee4d94e157
commit ee4d94e157
parent 59722acf76
7 changed files with 456 additions and 0 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@ -11,7 +11,9 @@ dependencies = [

 [project.optional-dependencies]
 dev = [
+    "hydra-core>=1.3.2",
    "matplotlib>=3.10.7",
+    "memray>=1.19.1",
    "optuna==4.5.0",
    "torch==2.9.0",
    "torchdata==0.7.1",