wip

pyproject.toml

@@ -8,6 +8,7 @@ dependencies = [
     "accelerate>=1.4.0",
     "activation-store",
     "autoawq>=0.2.6",
+    "beartype>=0.20.2",
     "datasets>=3.3.2",
     "einops>=0.8.1",
     "jaxtyping>=0.2.38",

research_journal.md

@@ -0,0 +1,3 @@
+
+# 2025-05-02 22:14:52
+

uv.lock

@@ -30,6 +30,7 @@ version = "0.1.0"
 source = { editable = "../../elk/cache_transformer_acts" }
 dependencies = [
     { name = "baukit" },
+    { name = "beartype" },
     { name = "datasets" },
     { name = "einops" },
     { name = "jaxtyping" },
@@ -44,6 +45,7 @@ dependencies = [
 [package.metadata]
 requires-dist = [
     { name = "baukit", git = "https://github.com/davidbau/baukit.git?rev=9d51abd51ebf29769aecc38c4cbef459b731a36e" },
+    { name = "beartype", specifier = ">=0.20.2" },
     { name = "datasets", specifier = ">=3.3.0" },
     { name = "einops", specifier = ">=0.8.1" },
     { name = "jaxtyping", specifier = ">=0.2.38" },
@@ -231,6 +233,15 @@ dependencies = [
     { name = "torchvision" },
 ]
 
+[[package]]
+name = "beartype"
+version = "0.20.2"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/8a/99/eaf0847301a6ca2042927c83f8cc72a29c153132ce49d9a720fe121440ba/beartype-0.20.2.tar.gz", hash = "sha256:38c60c065ad99364a8c767e8a0e71ba8263d467b91414ed5dcffb7758a2e8079", size = 1410390 }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/78/05/536d025b3e17cf938f836665dde32e86f65ee76acd0ae14e22bda6aee274/beartype-0.20.2-py3-none-any.whl", hash = "sha256:5171a91ecf01438a59884f0cde37d2d5da2c992198b53d6ba31db3940f47ff04", size = 1161292 },
+]
+
 [[package]]
 name = "certifi"
 version = "2025.1.31"
@@ -537,6 +548,7 @@ dependencies = [
     { name = "accelerate" },
     { name = "activation-store" },
     { name = "autoawq" },
+    { name = "beartype" },
     { name = "datasets" },
     { name = "einops" },
     { name = "jaxtyping" },
@@ -560,6 +572,7 @@ requires-dist = [
     { name = "accelerate", specifier = ">=1.4.0" },
     { name = "activation-store", editable = "../../elk/cache_transformer_acts" },
     { name = "autoawq", specifier = ">=0.2.6" },
+    { name = "beartype", specifier = ">=0.20.2" },
     { name = "datasets", specifier = ">=3.3.2" },
     { name = "einops", specifier = ">=0.8.1" },
     { name = "jaxtyping", specifier = ">=0.2.38" },