[FEATURE REQUEST] - Adding functionality for more tools for Arrakis

[yash-srivastava19]

In the first version of Arrakis, the following tools are supported(with built in function) :

Attention Head Composition
Attention Tools
Causal Tracing Intervention
Knowledge Graph Extractor
Knowledge Prober
Logit Attribution
LogitLens
Model Surgery
Read Write Heads
Residual Decomposer
Residual Tools
Sparsity Analyzer
Superposition Disentangler

Open a PR with another tool which are useful for MI. Before that, read the contributing guidelines and the pull request template(in the .github folder)