[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)