Arrakis’s documentation!# Contents: Introduction Attention Head Composition Attention Tools Causal Tracing Intervention Feature Visualization Knowledge Graph Extractor Knowledge Prober Logit Attribution LogitLens Model Surgery Read Write Heads Residual Decomposer Residual Tools Sparsity Analyzer Superposition Disentangler AttentionHeadComposition# AttentionTools# CausalTracingIntervention# KnowledgeGraphExtractor# KnowledgeProber# LogitAttribution# LogitLens# class core_arrakis.logit_lens.LogitLens(model)# Bases: BaseInterpretabilityTool Analyzes the logit lens of the model. layer_attributions(input_ids, target_idx)# Computes the attributions of the layers. logit_lens(input_ids, target_idx, layer_idx)# Computes the logit lens of the model. ModelSurgery# ReadWriteHeads# ResidualDecomposer# ResidualTools# SparsityAnalyzer# SuperpositionDisentangler#