Proposal
PR #466 was created with the idea of adding a specific set of functions or utilities that could do the DLA process in a single call. The PR has been stale for two years now, but the idea behind it is sound.
Review the PR and create a new tool in transformer_lens/tools/analysis/direct_logit_attribution.py that handles this for the new TransformerBridge system.
Proposal
PR #466 was created with the idea of adding a specific set of functions or utilities that could do the DLA process in a single call. The PR has been stale for two years now, but the idea behind it is sound.
Review the PR and create a new tool in
transformer_lens/tools/analysis/direct_logit_attribution.pythat handles this for the newTransformerBridgesystem.