replicated mecc

2025-07-31 01:12:53 -04:00
parent 1a0425d549
commit 5a0c479c1e
21 changed files with 1304 additions and 54 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,6 +10,9 @@ dependencies = [
    "jax[cuda12]>=0.7.0",
    "jupyter>=1.1.1",
    "matplotlib>=3.10.3",
+    "networkx>=3.5",
+    "optax>=0.2.5",
+    "scikit-learn>=1.7.1",
    "seaborn>=0.13.2",
    "tqdm>=4.67.1",
-]
+]