What problem does it solve?
Streamslining loading, configuring, and surgically modifying models for experiment-ready setups by reducing boilerplate around loaders, hooks, and ablation utilities.
Core Features & Use Cases
- Model Loading: Load models from HuggingFace Hub or local checkpoints with correct config and dtype handling.
- Architecture Discovery: Generate architecture reports (model-architecture.json) and enumerate layers, parameters, and components.
- Hook Infrastructure: Attach forward hooks to extract intermediate activations and support controlled ablation.
- Ablation & Fine-tuning: Provide zero/mean activation ablation, activation patching, and reversible state changes for safe experiments.
- Comparison Tools: Enable representation similarity analyses (CKA/RSA) across models and checkpoints.
- Hydra Configs: Output Hydra YAMLs for model loading and experiment pipelines.
Quick Start
Load a pretrained model, attach a hook to an attention layer, run a test input to collect activations, and generate the architecture report.