LICENSE
README.md
pyproject.toml
e2e_sae/__init__.py
e2e_sae/data.py
e2e_sae/hooks.py
e2e_sae/loader.py
e2e_sae/log.py
e2e_sae/losses.py
e2e_sae/metrics.py
e2e_sae/plotting.py
e2e_sae/settings.py
e2e_sae/types.py
e2e_sae/utils.py
e2e_sae.egg-info/PKG-INFO
e2e_sae.egg-info/SOURCES.txt
e2e_sae.egg-info/dependency_links.txt
e2e_sae.egg-info/requires.txt
e2e_sae.egg-info/top_level.txt
e2e_sae/models/__init__.py
e2e_sae/models/mlp.py
e2e_sae/models/sparsifiers.py
e2e_sae/models/transformers.py
e2e_sae/scripts/__init__.py
e2e_sae/scripts/upload_hf_dataset.py
tests/test_mlp.py
tests/test_sae.py
tests/test_train_tlens_saes.py
tests/test_transformer.py
tests/test_utils.py