README.md
pyproject.toml
omnilens/__init__.py
omnilens.egg-info/PKG-INFO
omnilens.egg-info/SOURCES.txt
omnilens.egg-info/dependency_links.txt
omnilens.egg-info/requires.txt
omnilens.egg-info/top_level.txt
omnilens/core/__init__.py
omnilens/core/attention_wrap.py
omnilens/core/block_wrap.py
omnilens/core/cache.py
omnilens/core/tapped_model.py
omnilens/methods/__init__.py
omnilens/methods/activation_patching.py
omnilens/methods/logit_lens.py
omnilens/methods/probe.py
omnilens/methods/sae.py
omnilens/methods/steering.py
omnilens/methods/transcoder.py
omnilens/methods/xray.py
omnilens/registry/__init__.py
omnilens/registry/auto_detect.py
omnilens/registry/loader.py
omnilens/registry/architectures/bloom.yaml
omnilens/registry/architectures/falcon.yaml
omnilens/registry/architectures/gemma.yaml
omnilens/registry/architectures/gemma2.yaml
omnilens/registry/architectures/gemma3.yaml
omnilens/registry/architectures/gpt2.yaml
omnilens/registry/architectures/gpt_neox.yaml
omnilens/registry/architectures/gptj.yaml
omnilens/registry/architectures/llama.yaml
omnilens/registry/architectures/mamba.yaml
omnilens/registry/architectures/mistral.yaml
omnilens/registry/architectures/opt.yaml
omnilens/registry/architectures/phi.yaml
omnilens/registry/architectures/phi3.yaml
omnilens/registry/architectures/qwen2.yaml
omnilens/registry/architectures/qwen3.yaml
omnilens/registry/architectures/rwkv.yaml
omnilens/registry/architectures/stablelm.yaml
omnilens/viz/__init__.py
omnilens/viz/plots.py
tests/test_probe.py
tests/test_sae.py
tests/test_steering.py
tests/test_tapped_model.py
tests/test_transcoder.py