feat: add get_act_patch_direct_path for head-to-head circuit analysis#1396

Merged

jlarson4 merged 9 commits into

TransformerLensOrg:devfrom

mukund1985:feat/direct-path-patching

Jun 22, 2026

Commits on Jun 19, 2026

feat: add get_act_patch_direct_path for head-to-head circuit analysis
mukund1985
committed
style: apply black + isort formatting (line-length=100)
mukund1985
committed
fix: remove unused imports, add type: ignore for mypy, clean up demo import
mukund1985
committed
fix: add type: ignore[index] on W_O[h] indexing for mypy
mukund1985
committed
fix: address reviewer feedback — TransformerBridge support, fold_ln guard, independent test, notebook demo
mukund1985
committed
fix: check fold_ln for TransformerBridge via .weight attribute
mukund1985
committed
refactor: move to tools/analysis/, logit-diff metric, TestCheckFoldLn, fix _check_fold_ln tensor bug
mukund1985
committed
fix: loosen _check_fold_ln type hint to Any for beartype compatibility
mukund1985
committed
style: black formatting on test_direct_path_patching.py
mukund1985
committed