Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | R | S | T | U | V | W _ __add__() (tdhook.hooks.MultiHookHandle method) __call__() (tdhook.attribution.lrp_helpers.rules.BaseRuleMapper method) (tdhook.attribution.lrp_helpers.rules.ParamModifier method) (tdhook.metrics.InfidelityMetric method) (tdhook.metrics.SensitivityMetric method) __enter__() (tdhook.attribution.lrp_helpers.rules.RemovableRuleHandle method) (tdhook.contexts.HookingContext method) (tdhook.contexts.HookingContextWithCache method) (tdhook.hooks.MultiHookHandle method) (tdhook.modules.HookedModuleRun method) (tdhook.weights.pruning.PruningContext method) __exit__() (tdhook.attribution.lrp_helpers.rules.RemovableRuleHandle method) (tdhook.contexts.HookingContext method) (tdhook.hooks.MultiHookHandle method) (tdhook.modules.HookedModuleRun method) (tdhook.weights.pruning.PruningContext method) __instancecheck__() (tdhook._types.UnraveledMeta method) (tdhook.attribution.lrp_helpers.types.SubclassMeta method) __repr__() (tdhook.latent.CkaEstimator method) (tdhook.latent.dimension_estimation.ca_pca.CaPcaDimensionEstimator method) (tdhook.latent.dimension_estimation.CaPcaDimensionEstimator method) (tdhook.latent.dimension_estimation.local_knn.LocalKnnDimensionEstimator method) (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator method) (tdhook.latent.dimension_estimation.LocalKnnDimensionEstimator method) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator method) (tdhook.latent.dimension_estimation.twonn.TwoNnDimensionEstimator method) (tdhook.latent.dimension_estimation.TwoNnDimensionEstimator method) (tdhook.latent.InformationImbalanceEstimator method) (tdhook.latent.LocalKnnDimensionEstimator method) (tdhook.latent.representation_similarity.cka.CkaEstimator method) (tdhook.latent.representation_similarity.CkaEstimator method) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator method) (tdhook.latent.representation_similarity.InformationImbalanceEstimator method) (tdhook.latent.TwoNnDimensionEstimator method) (tdhook.modules.FunctionModule method) (tdhook.modules.HookedModule method) (tdhook.modules.IntermediateKeysCleaner method) (tdhook.modules.ModuleCall method) (tdhook.modules.ModuleCallWithCache method) (tdhook.modules.PGDModule method) __subclass__ (tdhook.attribution.lrp_helpers.types.Activation attribute) (tdhook.attribution.lrp_helpers.types.AvgPool attribute) (tdhook.attribution.lrp_helpers.types.BatchNorm attribute) (tdhook.attribution.lrp_helpers.types.Convolution attribute) (tdhook.attribution.lrp_helpers.types.ConvolutionStandard attribute) (tdhook.attribution.lrp_helpers.types.ConvolutionTranspose attribute) (tdhook.attribution.lrp_helpers.types.Linear attribute) (tdhook.attribution.lrp_helpers.types.MaxPool attribute) __subclasscheck__() (tdhook.attribution.lrp_helpers.types.SubclassMeta method) _abort_enter() (tdhook.contexts.HookingContext method) _absolute (tdhook.attribution.grad_cam.GradCAM attribute) (tdhook.attribution.GradCAM attribute) (tdhook.attribution.Saliency attribute) (tdhook.attribution.saliency.Saliency attribute) _adapters (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) _additional_init_keys (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _additional_keys (tdhook.latent.Probing attribute) (tdhook.latent.probing.context.Probing attribute) (tdhook.latent.probing.Probing attribute) _allow_overwrite (tdhook.latent.ProbeManager attribute) (tdhook.latent.probing.managers.ProbeManager attribute) (tdhook.latent.probing.ProbeManager attribute) _alpha (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) (tdhook.modules.PGDModule attribute) _alphas (tdhook.attribution.integrated_gradients.IntegratedGradients attribute) (tdhook.attribution.IntegratedGradients attribute) _amount_to_prune (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) _apply_kwargs (tdhook.attribution.lrp_helpers.rules.Rule attribute) _artifact_keys() (tdhook.pipeline.Pipeline static method) _as_distance_input_dtype() (in module tdhook.latent.representation_similarity.information_imbalance) _ascent (tdhook.modules.PGDModule attribute) _attr_key (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _attribution_key (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) _attributor_fn() (tdhook.attribution.gradient_helpers.common.GradientAttribution method) (tdhook.attribution.gradient_helpers.GradientAttribution method) _baseline_key (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline attribute) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline attribute) _batch_size (tdhook.latent.probing.estimators.TorchEstimator attribute) _ca_pca() (in module tdhook.latent.dimension_estimation.ca_pca) _cache (tdhook.contexts.HookingContextWithCache attribute) (tdhook.hooks.CacheProxy attribute) (tdhook.modules.HookedModuleRun attribute) _cache_as_output (tdhook.modules.ModuleCallWithCache attribute) _cache_callback (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) (tdhook.latent.activation_patching.ActivationPatching attribute) (tdhook.latent.ActivationAddition attribute) (tdhook.latent.ActivationPatching attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) _cache_key (tdhook.modules.ModuleCallWithCache attribute) _cache_ref (tdhook.modules.ModuleCallWithCache attribute) _cached (tdhook.latent.BilinearProbe attribute) (tdhook.latent.probing.BilinearProbe attribute) (tdhook.latent.probing.managers.BilinearProbe attribute) _call() (tdhook.attribution.lrp_helpers.rules.BaseRuleMapper method) (tdhook.attribution.lrp_helpers.rules.EpsilonPlus method) _callback (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) _check_callback_signature() (tdhook.hooks.HookFactory static method) _check_hook_signature() (in module tdhook.hooks) _child_module() (tdhook.contexts.CompositeHookingContextFactory static method) _classes_to_hook (tdhook.hooks.MultiHookManager attribute) _classes_to_skip (tdhook.hooks.MultiHookManager attribute) _clean_intermediate_keys (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) (tdhook.latent.activation_patching.ActivationPatching attribute) (tdhook.latent.ActivationAddition attribute) (tdhook.latent.ActivationPatching attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) _clear_cache (tdhook.contexts.HookingContextWithCache attribute) _coef (tdhook.latent.MeanDifferenceClassifier attribute) (tdhook.latent.probing.estimators.MeanDifferenceClassifier attribute) (tdhook.latent.probing.MeanDifferenceClassifier attribute) _compute_convergence_delta (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline attribute) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline attribute) _compute_convergence_delta_fn() (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline method) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline method) _compute_metrics (tdhook.latent.ProbeManager attribute) (tdhook.latent.probing.managers.ProbeManager attribute) (tdhook.latent.probing.ProbeManager attribute) _compute_ranks_from_dist() (in module tdhook.latent.representation_similarity.information_imbalance) _compute_steering_vectors() (tdhook.latent.ActivationAddition method) (tdhook.latent.steering_vectors.ActivationAddition method) _contexts (tdhook.contexts.CompositeHookingContextFactory attribute) _create_pair_probe() (tdhook.latent.BilinearProbeManager method) (tdhook.latent.probing.BilinearProbeManager method) (tdhook.latent.probing.managers.BilinearProbeManager method) _data (tdhook.modules.HookedModuleRun attribute) _data_preprocess_callback (tdhook.latent.Probe attribute) (tdhook.latent.ProbeManager attribute) (tdhook.latent.probing.managers.Probe attribute) (tdhook.latent.probing.managers.ProbeManager attribute) (tdhook.latent.probing.Probe attribute) (tdhook.latent.probing.ProbeManager attribute) _decision_function() (tdhook.latent.MeanDifferenceClassifier method) (tdhook.latent.probing.estimators.MeanDifferenceClassifier method) (tdhook.latent.probing.MeanDifferenceClassifier method) _default_data_preprocess_callback() (tdhook.latent.Probe method) (tdhook.latent.probing.managers.Probe method) (tdhook.latent.probing.Probe method) _device (tdhook.latent.probing.estimators.TorchEstimator attribute) _dim_from_ca_pca() (in module tdhook.latent.dimension_estimation.ca_pca) _dim_from_eigenvalues_maxgap() (in module tdhook.latent.dimension_estimation.local_pca) _dim_from_eigenvalues_ratio() (in module tdhook.latent.dimension_estimation.local_pca) _directional_imbalance() (in module tdhook.latent.representation_similarity.information_imbalance) _directions (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) (tdhook.latent.Probing attribute) (tdhook.latent.probing.context.Probing attribute) (tdhook.latent.probing.Probing attribute) (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) _ensure_in_context() (tdhook.modules.HookedModuleRun method) _ensure_sklearn() (in module tdhook._optional_deps) _enter() (tdhook.contexts.HookingContext method) (tdhook.contexts.HookingContextWithCache method) (tdhook.weights.pruning.PruningContext method) _epochs (tdhook.latent.probing.estimators.TorchEstimator attribute) _estimator (tdhook.latent.Probe attribute) (tdhook.latent.probing.managers.Probe attribute) (tdhook.latent.probing.Probe attribute) _estimator_class (tdhook.latent.ProbeManager attribute) (tdhook.latent.probing.managers.ProbeManager attribute) (tdhook.latent.probing.ProbeManager attribute) _estimator_kwargs (tdhook.latent.ProbeManager attribute) (tdhook.latent.probing.managers.ProbeManager attribute) (tdhook.latent.probing.ProbeManager attribute) _estimators (tdhook.latent.ProbeManager attribute) (tdhook.latent.probing.managers.ProbeManager attribute) (tdhook.latent.probing.ProbeManager attribute) _fit_callback (tdhook.latent.Probe attribute) (tdhook.latent.probing.managers.Probe attribute) (tdhook.latent.probing.Probe attribute) _fit_metrics (tdhook.latent.ProbeManager attribute) (tdhook.latent.probing.managers.ProbeManager attribute) (tdhook.latent.probing.ProbeManager attribute) _flatten (tdhook.modules.ModuleCall attribute) (tdhook.modules.ModuleCallWithCache attribute) _grad_attr() (tdhook.attribution.grad_cam.GradCAM method) (tdhook.attribution.GradCAM method) (tdhook.attribution.gradient_helpers.common.GradientAttribution method) (tdhook.attribution.gradient_helpers.GradientAttribution method) (tdhook.attribution.guided_backpropagation.GuidedBackpropagation method) (tdhook.attribution.GuidedBackpropagation method) (tdhook.attribution.integrated_gradients.IntegratedGradients method) (tdhook.attribution.IntegratedGradients method) (tdhook.attribution.LRP method) (tdhook.attribution.lrp.LRP method) (tdhook.attribution.Saliency method) (tdhook.attribution.saliency.Saliency method) _grad_enabled (tdhook.modules.HookedModuleRun attribute) _grad_key (tdhook.modules.PGDModule attribute) _h1_key (tdhook.latent.BilinearProbe attribute) (tdhook.latent.probing.BilinearProbe attribute) (tdhook.latent.probing.managers.BilinearProbe attribute) _h2_key (tdhook.latent.BilinearProbe attribute) (tdhook.latent.probing.BilinearProbe attribute) (tdhook.latent.probing.managers.BilinearProbe attribute) _handle (tdhook.contexts.HookingContext attribute) _handles (tdhook.hooks.MultiHookHandle attribute) (tdhook.modules.HookedModuleRun attribute) _has_torchvision (in module tdhook.attribution.lrp_helpers.types) _hook (tdhook.contexts.HookingContext attribute) _hook_manager (tdhook.attribution.guided_backpropagation.GuidedBackpropagation attribute) (tdhook.attribution.GuidedBackpropagation attribute) (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) (tdhook.latent.Probing attribute) (tdhook.latent.probing.context.Probing attribute) (tdhook.latent.probing.Probing attribute) _hook_module() (tdhook.attribution.gradient_helpers.common.GradientAttribution method) (tdhook.attribution.gradient_helpers.GradientAttribution method) (tdhook.attribution.guided_backpropagation.GuidedBackpropagation method) (tdhook.attribution.GuidedBackpropagation method) (tdhook.contexts.CompositeHookingContextFactory method) (tdhook.contexts.HookingContextFactory method) (tdhook.latent.activation_caching.ActivationCaching method) (tdhook.latent.activation_patching.ActivationPatching method) (tdhook.latent.ActivationAddition method) (tdhook.latent.ActivationCaching method) (tdhook.latent.ActivationPatching method) (tdhook.latent.Probing method) (tdhook.latent.probing.context.Probing method) (tdhook.latent.probing.Probing method) (tdhook.latent.steering_vectors.ActivationAddition method) (tdhook.latent.steering_vectors.SteeringVectors method) (tdhook.latent.SteeringVectors method) (tdhook.weights.Adapters method) (tdhook.weights.adapters.Adapters method) _hooked_module (tdhook.contexts.HookingContext attribute) _hooked_module_class (tdhook.contexts.HookingContextFactory attribute) (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) (tdhook.weights.task_vectors.TaskVectors attribute) (tdhook.weights.TaskVectors attribute) _hooked_module_kwargs (tdhook.contexts.HookingContextFactory attribute) _hooking_context (tdhook.modules.HookedModule attribute) _hooking_context_class (tdhook.contexts.HookingContextFactory attribute) (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) (tdhook.weights.task_vectors.TaskVectors attribute) (tdhook.weights.TaskVectors attribute) _hooking_context_kwargs (tdhook.contexts.HookingContextFactory attribute) _importance_callback (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) _in_context (tdhook.contexts.HookingContext attribute) (tdhook.modules.HookedModuleRun attribute) _in_key (tdhook.modules.ModuleCall attribute) (tdhook.modules.ModuleCallWithCache attribute) _in_keys (tdhook.contexts.HookingContext attribute) _information_imbalance() (in module tdhook.latent.representation_similarity.information_imbalance) _init_attr_cache_in (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _init_attr_grads (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _init_attr_inputs (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _init_attr_targets (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _input_modules (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _intercept (tdhook.latent.MeanDifferenceClassifier attribute) (tdhook.latent.probing.estimators.MeanDifferenceClassifier attribute) (tdhook.latent.probing.MeanDifferenceClassifier attribute) _intermediate_keys (tdhook.modules.IntermediateKeysCleaner attribute) _key (tdhook.hooks.CacheProxy attribute) (tdhook.hooks.EarlyStoppingException attribute) _key_pattern (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) (tdhook.latent.Probing attribute) (tdhook.latent.probing.context.Probing attribute) (tdhook.latent.probing.Probing attribute) _key_to_probes (tdhook.latent.BilinearProbeManager attribute) (tdhook.latent.probing.BilinearProbeManager attribute) (tdhook.latent.probing.managers.BilinearProbeManager attribute) _keys() (in module tdhook.pipeline) _keys_conflict() (in module tdhook.pipeline) _linear_cka() (in module tdhook.latent.representation_similarity.cka) _local_knn() (in module tdhook.latent.dimension_estimation.local_knn) _local_pca() (in module tdhook.latent.dimension_estimation.local_pca) _loss_fn() (tdhook.latent.probing.estimators.TorchEstimator method) _lr (tdhook.latent.probing.estimators.TorchEstimator attribute) _managed_by_context_manager (tdhook.contexts.HookingContext attribute) _max_value (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) (tdhook.modules.PGDModule attribute) _method (tdhook.attribution.integrated_gradients.IntegratedGradients attribute) (tdhook.attribution.IntegratedGradients attribute) _min_value (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) (tdhook.modules.PGDModule attribute) _modify_fn (tdhook.attribution.lrp_helpers.rules.ParamModifier attribute) _module (tdhook.modules.HookedModuleRun attribute) _module_ref (tdhook.attribution.lrp_helpers.rules.RemovableRuleHandle attribute) _modules_to_attribute (tdhook.attribution.grad_cam.GradCAM attribute) (tdhook.attribution.GradCAM attribute) _modules_to_maximise (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) _modules_to_patch (tdhook.latent.activation_patching.ActivationPatching attribute) (tdhook.latent.ActivationPatching attribute) _modules_to_prune (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) _modules_to_steer (tdhook.latent.ActivationAddition attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) (tdhook.latent.steering_vectors.SteeringVectors attribute) (tdhook.latent.SteeringVectors attribute) _multiply_by_inputs (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline attribute) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline attribute) (tdhook.attribution.guided_backpropagation.GuidedBackpropagation attribute) (tdhook.attribution.GuidedBackpropagation attribute) (tdhook.attribution.Saliency attribute) (tdhook.attribution.saliency.Saliency attribute) _multiply_by_inputs_fn() (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline method) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline method) _n_steps (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) (tdhook.attribution.integrated_gradients.IntegratedGradients attribute) (tdhook.attribution.IntegratedGradients attribute) (tdhook.modules.PGDModule attribute) _name (tdhook.modules.HookedModuleRun attribute) _negative_key (tdhook.latent.ActivationAddition attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) _normalize (tdhook.latent.MeanDifferenceClassifier attribute) (tdhook.latent.probing.estimators.MeanDifferenceClassifier attribute) (tdhook.latent.probing.MeanDifferenceClassifier attribute) _num_classes (tdhook.latent.probing.estimators.TorchEstimator attribute) _old_weights (tdhook.weights.pruning.PruningContext attribute) _out_key (tdhook.modules.ModuleCall attribute) (tdhook.modules.ModuleCallWithCache attribute) _out_keys (tdhook.contexts.HookingContext attribute) _outer_cache (tdhook.modules.HookedModuleRun attribute) _output_grad_callbacks (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _pair_probes (tdhook.latent.BilinearProbeManager attribute) (tdhook.latent.probing.BilinearProbeManager attribute) (tdhook.latent.probing.managers.BilinearProbeManager attribute) _pairs (tdhook.latent.BilinearProbeManager attribute) (tdhook.latent.probing.BilinearProbeManager attribute) (tdhook.latent.probing.managers.BilinearProbeManager attribute) _patch_fn (tdhook.latent.activation_patching.ActivationPatching attribute) (tdhook.latent.ActivationPatching attribute) _patch_key (tdhook.latent.activation_patching.ActivationPatching attribute) (tdhook.latent.ActivationPatching attribute) _path() (in module tdhook.pipeline) _pattern (tdhook.hooks.MultiHookManager attribute) _perturb_data() (tdhook.metrics.InfidelityMetric method) (tdhook.metrics.SensitivityMetric method) _pgd_step() (tdhook.modules.PGDModule method) _positive_key (tdhook.latent.ActivationAddition attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) _pre_factories (tdhook.contexts.HookingContext attribute) _predict_callback (tdhook.latent.Probe attribute) (tdhook.latent.probing.managers.Probe attribute) (tdhook.latent.probing.Probe attribute) _predict_metrics (tdhook.latent.ProbeManager attribute) (tdhook.latent.probing.managers.ProbeManager attribute) (tdhook.latent.probing.ProbeManager attribute) _prepare (tdhook.contexts.HookingContext attribute) _prepare_module() (tdhook.attribution.activation_maximisation.ActivationMaximisation method) (tdhook.attribution.gradient_helpers.common.GradientAttribution method) (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline method) (tdhook.attribution.gradient_helpers.GradientAttribution method) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline method) (tdhook.attribution.LRP method) (tdhook.attribution.lrp.LRP method) (tdhook.contexts.CompositeHookingContextFactory method) (tdhook.contexts.HookingContextFactory method) (tdhook.latent.activation_patching.ActivationPatching method) (tdhook.latent.ActivationAddition method) (tdhook.latent.ActivationPatching method) (tdhook.latent.steering_vectors.ActivationAddition method) (tdhook.weights.Pruning method) (tdhook.weights.pruning.Pruning method) _prepared_children (tdhook.contexts.CompositeHookingContextFactory attribute) _probe_factory (tdhook.latent.Probing attribute) (tdhook.latent.probing.context.Probing attribute) (tdhook.latent.probing.Probing attribute) _reduce_baselines_fn() (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline method) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline method) (tdhook.attribution.integrated_gradients.IntegratedGradients method) (tdhook.attribution.IntegratedGradients method) _ref (tdhook.hooks.MutableWeakRef attribute) _reg_exp (tdhook.hooks.MultiHookManager attribute) _register_inputs_fn() (tdhook.attribution.gradient_helpers.common.GradientAttribution method) (tdhook.attribution.gradient_helpers.GradientAttribution method) _relative (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) (tdhook.latent.Probing attribute) (tdhook.latent.probing.context.Probing attribute) (tdhook.latent.probing.Probing attribute) (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) _relative_path (tdhook.modules.HookedModule attribute) (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) _resolve_k() (in module tdhook.latent.dimension_estimation.local_knn) _restore (tdhook.contexts.HookingContext attribute) _restore_module() (tdhook.attribution.LRP method) (tdhook.attribution.lrp.LRP method) (tdhook.contexts.CompositeHookingContextFactory method) (tdhook.contexts.HookingContextFactory method) _rule (tdhook.attribution.lrp_helpers.rules.RemovableRuleHandle attribute) _rule_mapper (tdhook.attribution.LRP attribute) (tdhook.attribution.lrp.LRP attribute) (tdhook.attribution.lrp_helpers.rules.BaseRuleMapper attribute) _rules (tdhook.attribution.lrp_helpers.rules.BaseRuleMapper attribute) _run() (tdhook.latent.BilinearProbe method) (tdhook.latent.probing.BilinearProbe method) (tdhook.latent.probing.managers.BilinearProbe method) _run_callback (tdhook.modules.HookedModuleRun attribute) _save_cache (tdhook.modules.HookedModuleRun attribute) _select_fn (tdhook.attribution.lrp_helpers.rules.ParamModifier attribute) _sep (tdhook.modules.HookedModuleRun attribute) _skip_modules (tdhook.attribution.LRP attribute) (tdhook.attribution.lrp.LRP attribute) (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) _spawn (tdhook.contexts.HookingContext attribute) _spawn_hooked_module() (tdhook.contexts.HookingContextFactory method) _stabilizer (tdhook.attribution.lrp_helpers.rules.BaseRuleMapper attribute) _stack (tdhook.contexts.HookingContext attribute) _steer_fn (tdhook.latent.steering_vectors.SteeringVectors attribute) (tdhook.latent.SteeringVectors attribute) _steer_key (tdhook.latent.ActivationAddition attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) _step_sizes (tdhook.attribution.integrated_gradients.IntegratedGradients attribute) (tdhook.attribution.IntegratedGradients attribute) _target_modules (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _td (tdhook.hooks.TensorDictRef attribute) _td_fn (tdhook.modules.FunctionModule attribute) _td_module (tdhook.modules.ModuleCall attribute) (tdhook.modules.ModuleCallWithCache attribute) (tdhook.modules.PGDModule attribute) _twonn() (in module tdhook.latent.dimension_estimation.twonn) _use_inputs (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _use_nested_keys (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) _use_outputs (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _use_sign (tdhook.modules.PGDModule attribute) _validate_capabilities() (tdhook.contexts.CompositeHookingContextFactory method) _validate_inputs() (in module tdhook.latent.representation_similarity.cka) (in module tdhook.latent.representation_similarity.information_imbalance) _validate_static() (tdhook.pipeline.Pipeline method) _verbose (tdhook.latent.probing.estimators.TorchEstimator attribute) _waiting_active (tdhook.latent.BilinearProbe attribute) (tdhook.latent.probing.BilinearProbe attribute) (tdhook.latent.probing.managers.BilinearProbe attribute) _warn_on_missing_rule (tdhook.attribution.LRP attribute) (tdhook.attribution.lrp.LRP attribute) _weights (tdhook.weights.task_vectors.TaskVectorsModule attribute) _working_key (tdhook.modules.PGDModule attribute) A AbstractFunctionMeta (class in tdhook.attribution.lrp_helpers.rules) Activation (class in tdhook.attribution.lrp_helpers.types) ActivationAddition (class in tdhook.latent) (class in tdhook.latent.steering_vectors) ActivationCaching (class in tdhook.latent) (class in tdhook.latent.activation_caching) ActivationMaximisation (class in tdhook.attribution.activation_maximisation) ActivationPatching (class in tdhook.latent) (class in tdhook.latent.activation_patching) Adapters (class in tdhook.weights) (class in tdhook.weights.adapters) adapters (tdhook.weights.adapters.HookedModuleWithAdapters attribute) after_all() (tdhook.latent.BilinearProbe method) (tdhook.latent.BilinearProbeManager method) (tdhook.latent.probing.BilinearProbe method) (tdhook.latent.probing.BilinearProbeManager method) (tdhook.latent.probing.managers.BilinearProbe method) (tdhook.latent.probing.managers.BilinearProbeManager method) AHQKVRule (class in tdhook.attribution.lrp_helpers.rules) alpha (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator attribute) AlphaBetaRule (class in tdhook.attribution.lrp_helpers.rules) alphas (tdhook.weights.task_vectors.TaskVectorsContext attribute) approximation_parameters() (in module tdhook.attribution.gradient_helpers.helpers) artifacts (tdhook.pipeline.PipelineResult attribute) AvgPool (class in tdhook.attribution.lrp_helpers.types) B backward() (tdhook.attribution.lrp_helpers.rules.AHQKVRule static method) (tdhook.attribution.lrp_helpers.rules.AlphaBetaRule static method) (tdhook.attribution.lrp_helpers.rules.EpsilonRule static method) (tdhook.attribution.lrp_helpers.rules.IgnoreRule static method) (tdhook.attribution.lrp_helpers.rules.LayerNormRule static method) (tdhook.attribution.lrp_helpers.rules.PassRule static method) (tdhook.attribution.lrp_helpers.rules.PseudoIdentityRule static method) (tdhook.attribution.lrp_helpers.rules.Rule static method) (tdhook.attribution.lrp_helpers.rules.SoftmaxEpsilonRule static method) (tdhook.attribution.lrp_helpers.rules.StopRule static method) (tdhook.attribution.lrp_helpers.rules.UniformEpsilonRule static method) (tdhook.attribution.lrp_helpers.rules.UniformRule static method) (tdhook.attribution.lrp_helpers.rules.WSquareRule static method) BaseRuleMapper (class in tdhook.attribution.lrp_helpers.rules) BatchNorm (class in tdhook.attribution.lrp_helpers.types) before_all() (tdhook.latent.BilinearProbe method) (tdhook.latent.BilinearProbeManager method) (tdhook.latent.probing.BilinearProbe method) (tdhook.latent.probing.BilinearProbeManager method) (tdhook.latent.probing.managers.BilinearProbe method) (tdhook.latent.probing.managers.BilinearProbeManager method) bilinear (tdhook.latent.probing.estimators.BilinearEstimator attribute) BilinearEstimator (class in tdhook.latent.probing.estimators) BilinearProbe (class in tdhook.latent) (class in tdhook.latent.probing) (class in tdhook.latent.probing.managers) BilinearProbeManager (class in tdhook.latent) (class in tdhook.latent.probing) (class in tdhook.latent.probing.managers) C cache (tdhook.contexts.HookingContextWithCache property) (tdhook.modules.HookedModuleRun property) cache_ref (tdhook.modules.ModuleCallWithCache property) CacheProxy (class in tdhook.hooks) CaPcaDimensionEstimator (class in tdhook.latent.dimension_estimation) (class in tdhook.latent.dimension_estimation.ca_pca) CkaEstimator (class in tdhook.latent) (class in tdhook.latent.representation_similarity) (class in tdhook.latent.representation_similarity.cka) clear() (tdhook.contexts.HookingContextWithCache method) coef_ (tdhook.latent.MeanDifferenceClassifier property) (tdhook.latent.probing.estimators.MeanDifferenceClassifier property) (tdhook.latent.probing.MeanDifferenceClassifier property) CompositeHookingContextFactory (class in tdhook.contexts) compute_alpha() (tdhook.weights.task_vectors.TaskVectorsContext method) Convolution (class in tdhook.attribution.lrp_helpers.types) ConvolutionStandard (class in tdhook.attribution.lrp_helpers.types) ConvolutionTranspose (class in tdhook.attribution.lrp_helpers.types) criterion (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator attribute) D default_classes_to_hook (tdhook.latent.Probing attribute) (tdhook.latent.probing.context.Probing attribute) (tdhook.latent.probing.Probing attribute) default_classes_to_skip (tdhook.latent.Probing attribute) (tdhook.latent.probing.context.Probing attribute) (tdhook.latent.probing.Probing attribute) default_skip() (tdhook.attribution.LRP static method) (tdhook.attribution.lrp.LRP static method) (tdhook.weights.Pruning static method) (tdhook.weights.pruning.Pruning static method) dim (tdhook.attribution.lrp_helpers.layers.Sum attribute) DimsConfig (class in tdhook.attribution.grad_cam) DIRECTION_TO_PARAMS (in module tdhook.hooks) DIRECTION_TO_RETURN (in module tdhook.hooks) DIRECTION_TO_RETURN_INDEX (in module tdhook.hooks) DIRECTION_TO_TYPE (in module tdhook.hooks) disable() (tdhook.contexts.HookingContext method) disable_context() (tdhook.modules.HookedModule method) disable_context_hooks() (tdhook.modules.HookedModule method) disable_hooks() (tdhook.contexts.HookingContext method) E EarlyStoppingException effects (tdhook.pipeline.Stage attribute) (tdhook.pipeline.StageResult attribute) eps (tdhook.latent.CkaEstimator attribute) (tdhook.latent.dimension_estimation.ca_pca.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_knn.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.twonn.TwoNnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.TwoNnDimensionEstimator attribute) (tdhook.latent.LocalKnnDimensionEstimator attribute) (tdhook.latent.representation_similarity.cka.CkaEstimator attribute) (tdhook.latent.representation_similarity.CkaEstimator attribute) (tdhook.latent.TwoNnDimensionEstimator attribute) epsilon (tdhook.attribution.lrp_helpers.rules.EpsilonRule property) EpsilonPlus (class in tdhook.attribution.lrp_helpers.rules) EpsilonRule (class in tdhook.attribution.lrp_helpers.rules) estimators (tdhook.latent.ProbeManager property) (tdhook.latent.probing.managers.ProbeManager property) (tdhook.latent.probing.ProbeManager property) F factory (tdhook.pipeline.MethodStage attribute) feature_sum_dims (tdhook.attribution.grad_cam.DimsConfig attribute) fit() (tdhook.latent.MeanDifferenceClassifier method) (tdhook.latent.probing.estimators.MeanDifferenceClassifier method) (tdhook.latent.probing.estimators.TorchEstimator method) (tdhook.latent.probing.MeanDifferenceClassifier method) fit_metrics (tdhook.latent.ProbeManager property) (tdhook.latent.probing.managers.ProbeManager property) (tdhook.latent.probing.ProbeManager property) FlatRule (class in tdhook.attribution.lrp_helpers.rules) flatten_select_reshape_call() (in module tdhook.modules) forward() (tdhook.attribution.lrp_helpers.layers.Sum method) (tdhook.attribution.lrp_helpers.rules.AHQKVRule static method) (tdhook.attribution.lrp_helpers.rules.AlphaBetaRule static method) (tdhook.attribution.lrp_helpers.rules.EpsilonRule static method) (tdhook.attribution.lrp_helpers.rules.IgnoreRule static method) (tdhook.attribution.lrp_helpers.rules.LayerNormRule static method) (tdhook.attribution.lrp_helpers.rules.PassRule static method) (tdhook.attribution.lrp_helpers.rules.PseudoIdentityRule static method) (tdhook.attribution.lrp_helpers.rules.Rule static method) (tdhook.attribution.lrp_helpers.rules.StopRule static method) (tdhook.attribution.lrp_helpers.rules.UniformRule static method) (tdhook.attribution.lrp_helpers.rules.WSquareRule static method) (tdhook.latent.CkaEstimator method) (tdhook.latent.dimension_estimation.ca_pca.CaPcaDimensionEstimator method) (tdhook.latent.dimension_estimation.CaPcaDimensionEstimator method) (tdhook.latent.dimension_estimation.local_knn.LocalKnnDimensionEstimator method) (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator method) (tdhook.latent.dimension_estimation.LocalKnnDimensionEstimator method) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator method) (tdhook.latent.dimension_estimation.twonn.TwoNnDimensionEstimator method) (tdhook.latent.dimension_estimation.TwoNnDimensionEstimator method) (tdhook.latent.InformationImbalanceEstimator method) (tdhook.latent.LinearEstimator method) (tdhook.latent.LocalKnnDimensionEstimator method) (tdhook.latent.LowRankBilinearEstimator method) (tdhook.latent.probing.estimators.BilinearEstimator method) (tdhook.latent.probing.estimators.LinearEstimator method) (tdhook.latent.probing.estimators.LowRankBilinearEstimator method) (tdhook.latent.probing.LinearEstimator method) (tdhook.latent.probing.LowRankBilinearEstimator method) (tdhook.latent.representation_similarity.cka.CkaEstimator method) (tdhook.latent.representation_similarity.CkaEstimator method) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator method) (tdhook.latent.representation_similarity.InformationImbalanceEstimator method) (tdhook.latent.TwoNnDimensionEstimator method) (tdhook.modules.FunctionModule method) (tdhook.modules.HookedModule method) (tdhook.modules.IntermediateKeysCleaner method) (tdhook.modules.ModuleCall method) (tdhook.modules.ModuleCallWithCache method) (tdhook.modules.PGDModule method) from_modifiers() (tdhook.attribution.lrp_helpers.rules.ParamModifier static method) from_module() (tdhook.modules.HookedModule class method) FunctionModule (class in tdhook.modules) G gauss_legendre_builders() (in module tdhook.attribution.gradient_helpers.helpers) get() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) get_best_device() (in module tdhook.modules) get_control_adequacy (tdhook.weights.task_vectors.TaskVectorsContext attribute) get_forget_vector() (tdhook.weights.task_vectors.TaskVectorsModule method) get_grad() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) get_grad_output() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) get_input() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) get_task_vector() (tdhook.weights.task_vectors.TaskVectorsModule method) get_test_accuracy (tdhook.weights.task_vectors.TaskVectorsContext attribute) get_weights() (tdhook.weights.task_vectors.TaskVectorsModule method) GradCAM (class in tdhook.attribution) (class in tdhook.attribution.grad_cam) GradientAttribution (class in tdhook.attribution.gradient_helpers) (class in tdhook.attribution.gradient_helpers.common) GradientAttributionWithBaseline (class in tdhook.attribution.gradient_helpers) (class in tdhook.attribution.gradient_helpers.common) GuidedBackpropagation (class in tdhook.attribution) (class in tdhook.attribution.guided_backpropagation) H HookDirection (in module tdhook.hooks) HookedModule (class in tdhook.modules) HookedModuleRun (class in tdhook.modules) HookedModuleWithAdapters (class in tdhook.weights.adapters) HookFactory (class in tdhook.hooks) HookGroup (in module tdhook.contexts) hooking_context (tdhook.modules.HookedModule property) HookingContext (class in tdhook.contexts) HookingContextFactory (class in tdhook.contexts) HookingContextWithCache (class in tdhook.contexts) I IgnoreRule (class in tdhook.attribution.lrp_helpers.rules) in_key (tdhook.latent.dimension_estimation.ca_pca.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_knn.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.twonn.TwoNnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.TwoNnDimensionEstimator attribute) (tdhook.latent.LocalKnnDimensionEstimator attribute) (tdhook.latent.TwoNnDimensionEstimator attribute) in_key_a (tdhook.latent.CkaEstimator attribute) (tdhook.latent.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.cka.CkaEstimator attribute) (tdhook.latent.representation_similarity.CkaEstimator attribute) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.InformationImbalanceEstimator attribute) in_key_b (tdhook.latent.CkaEstimator attribute) (tdhook.latent.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.cka.CkaEstimator attribute) (tdhook.latent.representation_similarity.CkaEstimator attribute) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.InformationImbalanceEstimator attribute) in_keys (tdhook.latent.CkaEstimator attribute) (tdhook.latent.dimension_estimation.ca_pca.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_knn.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.twonn.TwoNnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.TwoNnDimensionEstimator attribute) (tdhook.latent.InformationImbalanceEstimator attribute) (tdhook.latent.LocalKnnDimensionEstimator attribute) (tdhook.latent.representation_similarity.cka.CkaEstimator attribute) (tdhook.latent.representation_similarity.CkaEstimator attribute) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.InformationImbalanceEstimator attribute) (tdhook.latent.TwoNnDimensionEstimator attribute) (tdhook.modules.FunctionModule attribute) (tdhook.modules.IntermediateKeysCleaner attribute) (tdhook.modules.ModuleCall attribute) (tdhook.modules.ModuleCallWithCache attribute) (tdhook.modules.PGDModule attribute) incompatible_effects (tdhook.pipeline.Stage attribute) InfidelityMetric (class in tdhook.metrics) InformationImbalanceEstimator (class in tdhook.latent) (class in tdhook.latent.representation_similarity) (class in tdhook.latent.representation_similarity.information_imbalance) init_attr_targets_with_labels() (tdhook.attribution.integrated_gradients.IntegratedGradients static method) (tdhook.attribution.IntegratedGradients static method) IntegratedGradients (class in tdhook.attribution) (class in tdhook.attribution.integrated_gradients) intercept_ (tdhook.latent.MeanDifferenceClassifier property) (tdhook.latent.probing.estimators.MeanDifferenceClassifier property) (tdhook.latent.probing.MeanDifferenceClassifier property) IntermediateKeysCleaner (class in tdhook.modules) is_waiting (tdhook.latent.BilinearProbe property) (tdhook.latent.probing.BilinearProbe property) (tdhook.latent.probing.managers.BilinearProbe property) K k (tdhook.latent.dimension_estimation.ca_pca.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_knn.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator attribute) (tdhook.latent.LocalKnnDimensionEstimator attribute) kernel (tdhook.latent.CkaEstimator attribute) (tdhook.latent.representation_similarity.cka.CkaEstimator attribute) (tdhook.latent.representation_similarity.CkaEstimator attribute) key_pattern (tdhook.latent.activation_caching.ActivationCaching property) (tdhook.latent.ActivationCaching property) (tdhook.latent.BilinearProbeManager property) (tdhook.latent.Probing property) (tdhook.latent.probing.BilinearProbeManager property) (tdhook.latent.probing.context.Probing property) (tdhook.latent.probing.managers.BilinearProbeManager property) (tdhook.latent.probing.Probing property) L LayerNormRule (class in tdhook.attribution.lrp_helpers.rules) left (tdhook.attribution.gradient_helpers.helpers.Riemann attribute) Linear (class in tdhook.attribution.lrp_helpers.types) linear (tdhook.latent.LinearEstimator attribute) (tdhook.latent.probing.estimators.LinearEstimator attribute) (tdhook.latent.probing.LinearEstimator attribute) linear1 (tdhook.latent.LowRankBilinearEstimator attribute) (tdhook.latent.probing.estimators.LowRankBilinearEstimator attribute) (tdhook.latent.probing.LowRankBilinearEstimator attribute) linear2 (tdhook.latent.LowRankBilinearEstimator attribute) (tdhook.latent.probing.estimators.LowRankBilinearEstimator attribute) (tdhook.latent.probing.LowRankBilinearEstimator attribute) LinearEstimator (class in tdhook.latent) (class in tdhook.latent.probing) (class in tdhook.latent.probing.estimators) LocalKnnDimensionEstimator (class in tdhook.latent) (class in tdhook.latent.dimension_estimation) (class in tdhook.latent.dimension_estimation.local_knn) LocalPcaDimensionEstimator (class in tdhook.latent.dimension_estimation) (class in tdhook.latent.dimension_estimation.local_pca) LowRankBilinearEstimator (class in tdhook.latent) (class in tdhook.latent.probing) (class in tdhook.latent.probing.estimators) LRP (class in tdhook.attribution) (class in tdhook.attribution.lrp) M make_caching_hook() (tdhook.hooks.HookFactory static method) make_reading_hook() (tdhook.hooks.HookFactory static method) make_setting_hook() (tdhook.hooks.HookFactory static method) make_stopping_hook() (tdhook.hooks.HookFactory static method) MaxPool (class in tdhook.attribution.lrp_helpers.types) MeanDifferenceClassifier (class in tdhook.latent) (class in tdhook.latent.probing) (class in tdhook.latent.probing.estimators) merge_paths() (in module tdhook.hooks) MethodStage (class in tdhook.pipeline) middle (tdhook.attribution.gradient_helpers.helpers.Riemann attribute) model_in_keys (tdhook.pipeline.MethodStage attribute) model_out_keys (tdhook.pipeline.MethodStage attribute) module tdhook tdhook._optional_deps tdhook._types tdhook.attribution tdhook.attribution.activation_maximisation tdhook.attribution.grad_cam tdhook.attribution.gradient_helpers tdhook.attribution.gradient_helpers.common tdhook.attribution.gradient_helpers.helpers tdhook.attribution.guided_backpropagation tdhook.attribution.integrated_gradients tdhook.attribution.lrp tdhook.attribution.lrp_helpers tdhook.attribution.lrp_helpers.layers tdhook.attribution.lrp_helpers.rules tdhook.attribution.lrp_helpers.types tdhook.attribution.saliency tdhook.auto tdhook.contexts tdhook.hooks tdhook.latent tdhook.latent.activation_caching tdhook.latent.activation_patching tdhook.latent.dimension_estimation tdhook.latent.dimension_estimation._utils tdhook.latent.dimension_estimation.ca_pca tdhook.latent.dimension_estimation.local_knn tdhook.latent.dimension_estimation.local_pca tdhook.latent.dimension_estimation.twonn tdhook.latent.probing tdhook.latent.probing.context tdhook.latent.probing.estimators tdhook.latent.probing.managers tdhook.latent.representation_similarity tdhook.latent.representation_similarity.cka tdhook.latent.representation_similarity.information_imbalance tdhook.latent.steering_vectors tdhook.metrics tdhook.modules tdhook.pipeline tdhook.sources tdhook.weights tdhook.weights.adapters tdhook.weights.pruning tdhook.weights.task_vectors ModuleCall (class in tdhook.modules) ModuleCallWithCache (class in tdhook.modules) MultiHookHandle (class in tdhook.hooks) MultiHookManager (class in tdhook.hooks) MutableWeakRef (class in tdhook.hooks) N n_perturb_samples (tdhook.metrics.InfidelityMetric attribute) name (tdhook.pipeline.Stage attribute) (tdhook.pipeline.StageResult attribute) O out_key (tdhook.latent.CkaEstimator attribute) (tdhook.latent.dimension_estimation.ca_pca.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_knn.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.twonn.TwoNnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.TwoNnDimensionEstimator attribute) (tdhook.latent.LocalKnnDimensionEstimator attribute) (tdhook.latent.representation_similarity.cka.CkaEstimator attribute) (tdhook.latent.representation_similarity.CkaEstimator attribute) (tdhook.latent.TwoNnDimensionEstimator attribute) out_key_a_to_b (tdhook.latent.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.InformationImbalanceEstimator attribute) out_key_b_to_a (tdhook.latent.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.InformationImbalanceEstimator attribute) out_keys (tdhook.latent.CkaEstimator attribute) (tdhook.latent.dimension_estimation.ca_pca.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_knn.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.twonn.TwoNnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.TwoNnDimensionEstimator attribute) (tdhook.latent.InformationImbalanceEstimator attribute) (tdhook.latent.LocalKnnDimensionEstimator attribute) (tdhook.latent.representation_similarity.cka.CkaEstimator attribute) (tdhook.latent.representation_similarity.CkaEstimator attribute) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.InformationImbalanceEstimator attribute) (tdhook.latent.TwoNnDimensionEstimator attribute) (tdhook.modules.FunctionModule attribute) (tdhook.modules.IntermediateKeysCleaner attribute) (tdhook.modules.ModuleCall attribute) (tdhook.modules.PGDModule attribute) P p (tdhook.latent.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.InformationImbalanceEstimator attribute) ParamModifier (class in tdhook.attribution.lrp_helpers.rules) PassRule (class in tdhook.attribution.lrp_helpers.rules) pattern (tdhook.hooks.MultiHookManager property) perturb_radius (tdhook.metrics.SensitivityMetric attribute) PGDModule (class in tdhook.modules) Pipeline (class in tdhook.pipeline) PipelineKey (in module tdhook.pipeline) PipelineResult (class in tdhook.pipeline) predict() (tdhook.latent.MeanDifferenceClassifier method) (tdhook.latent.probing.estimators.MeanDifferenceClassifier method) (tdhook.latent.probing.estimators.TorchEstimator method) (tdhook.latent.probing.MeanDifferenceClassifier method) predict_metrics (tdhook.latent.ProbeManager property) (tdhook.latent.probing.managers.ProbeManager property) (tdhook.latent.probing.ProbeManager property) predict_proba() (tdhook.latent.MeanDifferenceClassifier method) (tdhook.latent.probing.estimators.MeanDifferenceClassifier method) (tdhook.latent.probing.MeanDifferenceClassifier method) prepare() (tdhook.contexts.HookingContextFactory method) Probe (class in tdhook.latent) (class in tdhook.latent.probing) (class in tdhook.latent.probing.context) (class in tdhook.latent.probing.managers) probe_factory() (tdhook.latent.BilinearProbeManager method) (tdhook.latent.ProbeManager method) (tdhook.latent.probing.BilinearProbeManager method) (tdhook.latent.probing.managers.BilinearProbeManager method) (tdhook.latent.probing.managers.ProbeManager method) (tdhook.latent.probing.ProbeManager method) ProbeManager (class in tdhook.latent) (class in tdhook.latent.probing) (class in tdhook.latent.probing.managers) Probing (class in tdhook.latent) (class in tdhook.latent.probing) (class in tdhook.latent.probing.context) provided_keys (tdhook.pipeline.Stage attribute) (tdhook.pipeline.StageResult attribute) Pruning (class in tdhook.weights) (class in tdhook.weights.pruning) PruningContext (class in tdhook.weights.pruning) PseudoIdentityRule (class in tdhook.attribution.lrp_helpers.rules) R raise_for_unconserved_rel_factory() (in module tdhook.attribution.lrp_helpers.rules) register() (tdhook.attribution.lrp_helpers.rules.IgnoreRule method) (tdhook.attribution.lrp_helpers.rules.Rule method) register_hook() (tdhook.hooks.MultiHookManager method) register_hook_to_module() (in module tdhook.hooks) register_submodule_hook() (tdhook.modules.HookedModule method) relative_path (tdhook.modules.HookedModule property) RemovableHandleProtocol (class in tdhook.hooks) RemovableRuleHandle (class in tdhook.attribution.lrp_helpers.rules) remove() (tdhook.attribution.lrp_helpers.rules.RemovableRuleHandle method) (tdhook.hooks.MultiHookHandle method) (tdhook.hooks.RemovableHandleProtocol method) required_keys (tdhook.pipeline.Stage attribute) requires_sklearn() (in module tdhook._optional_deps) RESERVED_KEYS (in module tdhook.pipeline) reserved_keys (tdhook.pipeline.Pipeline attribute) reset_estimators() (tdhook.latent.BilinearProbeManager method) (tdhook.latent.ProbeManager method) (tdhook.latent.probing.BilinearProbeManager method) (tdhook.latent.probing.managers.BilinearProbeManager method) (tdhook.latent.probing.managers.ProbeManager method) (tdhook.latent.probing.ProbeManager method) reset_metrics() (tdhook.latent.BilinearProbeManager method) (tdhook.latent.ProbeManager method) (tdhook.latent.probing.BilinearProbeManager method) (tdhook.latent.probing.managers.BilinearProbeManager method) (tdhook.latent.probing.managers.ProbeManager method) (tdhook.latent.probing.ProbeManager method) resolve() (tdhook.hooks.CacheProxy method) (tdhook.hooks.MutableWeakRef method) (tdhook.hooks.TensorDictRef method) resolve_submodule_path() (in module tdhook.hooks) restore() (tdhook.modules.HookedModule method) return_xy (tdhook.latent.dimension_estimation.twonn.TwoNnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.TwoNnDimensionEstimator attribute) (tdhook.latent.TwoNnDimensionEstimator attribute) Riemann (class in tdhook.attribution.gradient_helpers.helpers) riemann_builders() (in module tdhook.attribution.gradient_helpers.helpers) right (tdhook.attribution.gradient_helpers.helpers.Riemann attribute) Rule (class in tdhook.attribution.lrp_helpers.rules) run() (tdhook.modules.HookedModule method) (tdhook.pipeline.MethodStage method) (tdhook.pipeline.Pipeline method) (tdhook.pipeline.Stage method) (tdhook.pipeline.TransformStage method) S Saliency (class in tdhook.attribution) (class in tdhook.attribution.saliency) save() (tdhook.modules.HookedModuleRun method) save_grad() (tdhook.modules.HookedModuleRun method) save_grad_output() (tdhook.modules.HookedModuleRun method) save_input() (tdhook.modules.HookedModuleRun method) select_all() (tdhook.attribution.lrp_helpers.rules.ParamModifier static method) SensitivityMetric (class in tdhook.metrics) set() (tdhook.hooks.MutableWeakRef method) (tdhook.hooks.TensorDictRef method) (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) set_grad() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) set_grad_output() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) set_input() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) SoftmaxEpsilonRule (class in tdhook.attribution.lrp_helpers.rules) sorted_neighbors() (in module tdhook.latent.dimension_estimation._utils) stabilize() (in module tdhook.attribution.lrp_helpers.rules) Stage (class in tdhook.pipeline) StageResult (class in tdhook.pipeline) stages (tdhook.pipeline.Pipeline attribute) (tdhook.pipeline.PipelineResult attribute) state_dicts() (tdhook.attribution.lrp_helpers.rules.ParamModifier method) SteeringVectors (class in tdhook.latent) (class in tdhook.latent.steering_vectors) step() (tdhook.latent.BilinearProbe method) (tdhook.latent.Probe method) (tdhook.latent.probing.BilinearProbe method) (tdhook.latent.probing.context.Probe method) (tdhook.latent.probing.managers.BilinearProbe method) (tdhook.latent.probing.managers.Probe method) (tdhook.latent.probing.Probe method) stop() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) StopRule (class in tdhook.attribution.lrp_helpers.rules) SubclassMeta (class in tdhook.attribution.lrp_helpers.types) submodule_path_to_name() (in module tdhook.hooks) Sum (class in tdhook.attribution.lrp_helpers.layers) SUPPORTED_METHODS (in module tdhook.attribution.gradient_helpers.helpers) SUPPORTED_RIEMANN_METHODS (in module tdhook.attribution.gradient_helpers.helpers) T T (in module tdhook.hooks) TaskVectors (class in tdhook.weights) (class in tdhook.weights.task_vectors) TaskVectorsContext (class in tdhook.weights.task_vectors) TaskVectorsModule (class in tdhook.weights.task_vectors) tdhook module tdhook._optional_deps module tdhook._types module tdhook.attribution module tdhook.attribution.activation_maximisation module tdhook.attribution.grad_cam module tdhook.attribution.gradient_helpers module tdhook.attribution.gradient_helpers.common module tdhook.attribution.gradient_helpers.helpers module tdhook.attribution.guided_backpropagation module tdhook.attribution.integrated_gradients module tdhook.attribution.lrp module tdhook.attribution.lrp_helpers module tdhook.attribution.lrp_helpers.layers module tdhook.attribution.lrp_helpers.rules module tdhook.attribution.lrp_helpers.types module tdhook.attribution.saliency module tdhook.auto module tdhook.contexts module tdhook.hooks module tdhook.latent module tdhook.latent.activation_caching module tdhook.latent.activation_patching module tdhook.latent.dimension_estimation module tdhook.latent.dimension_estimation._utils module tdhook.latent.dimension_estimation.ca_pca module tdhook.latent.dimension_estimation.local_knn module tdhook.latent.dimension_estimation.local_pca module tdhook.latent.dimension_estimation.twonn module tdhook.latent.probing module tdhook.latent.probing.context module tdhook.latent.probing.estimators module tdhook.latent.probing.managers module tdhook.latent.representation_similarity module tdhook.latent.representation_similarity.cka module tdhook.latent.representation_similarity.information_imbalance module tdhook.latent.steering_vectors module tdhook.metrics module tdhook.modules module tdhook.pipeline module tdhook.sources module tdhook.weights module tdhook.weights.adapters module tdhook.weights.pruning module tdhook.weights.task_vectors module TensorDictRef (class in tdhook.hooks) TorchEstimator (class in tdhook.latent.probing.estimators) torchvision (in module tdhook.attribution.lrp_helpers.types) transform (tdhook.pipeline.TransformStage attribute) TransformStage (class in tdhook.pipeline) trapezoid (tdhook.attribution.gradient_helpers.helpers.Riemann attribute) TwoNnDimensionEstimator (class in tdhook.latent) (class in tdhook.latent.dimension_estimation) (class in tdhook.latent.dimension_estimation.twonn) U UniformEpsilonRule (class in tdhook.attribution.lrp_helpers.rules) UniformRule (class in tdhook.attribution.lrp_helpers.rules) UnraveledKey (class in tdhook._types) UnraveledMeta (class in tdhook._types) unregister() (tdhook.attribution.lrp_helpers.rules.IgnoreRule method) (tdhook.attribution.lrp_helpers.rules.Rule method) V validate() (tdhook.pipeline.Pipeline method) W weight_pooling_dims (tdhook.attribution.grad_cam.DimsConfig attribute) with_applied_vectors() (tdhook.weights.task_vectors.TaskVectorsModule method) WSquareRule (class in tdhook.attribution.lrp_helpers.rules)
