transformer_lens.tools package

Subpackages

Submodules

Module contents

TransformerLens tools package.

This package contains utilities and tools for working with TransformerLens, including the model registry for discovering compatible HuggingFace models.

Subpackages:

  • analysis: High-level interpretability analyses (e.g. Direct Logit Attribution)

  • model_registry: Tools for discovering and documenting supported models

Subpackages load lazily (PEP 562): analysis pulls in torch-heavy core modules, so an eager import here would create a cycle for core modules that consume model_registry data at import time.