sae-lens

View on PyPIReverse Dependencies (1)

2.1.3 sae_lens-2.1.3-py3-none-any.whl

Wheel Details

Project: sae-lens
Version: 2.1.3
Filename: sae_lens-2.1.3-py3-none-any.whl
Download: [link]
Size: 76236
MD5: ae73852453b90e67ee33eec9634ae9d3
SHA256: faef05a001f7c59b84d5b057edb3253c1ab8d7ade35f4506bee2e25514795ba7
Uploaded: 2024-05-15 20:52:12 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: sae-lens
Version: 2.1.3
Summary: Training and Analyzing Sparse Autoencoders (SAEs)
Author: Joseph Bloom
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Requires-Python: >=3.10,<4.0
Requires-Dist: automated-interpretability (<0.0.4,>=0.0.3)
Requires-Dist: babe (<0.0.8,>=0.0.7)
Requires-Dist: datasets (<3.0.0,>=2.17.1)
Requires-Dist: ipykernel (<7.0.0,>=6.29.2)
Requires-Dist: jupyter (<2.0.0,>=1.0.0)
Requires-Dist: mamba-lens (<0.0.5,>=0.0.4); extra == "mamba"
Requires-Dist: matplotlib (<4.0.0,>=3.8.3)
Requires-Dist: matplotlib-inline (<0.2.0,>=0.1.6)
Requires-Dist: mkdocs (<2.0.0,>=1.5.3)
Requires-Dist: mkdocs-autorefs (<2.0.0,>=1.0.1)
Requires-Dist: mkdocs-material (<10.0.0,>=9.5.15)
Requires-Dist: mkdocs-section-index (<0.4.0,>=0.3.8)
Requires-Dist: mkdocstrings (<0.25.0,>=0.24.1)
Requires-Dist: mkdocstrings-python (<2.0.0,>=1.9.0)
Requires-Dist: nbformat (<6.0.0,>=5.9.2)
Requires-Dist: nltk (<4.0.0,>=3.8.1)
Requires-Dist: plotly (<6.0.0,>=5.19.0)
Requires-Dist: plotly-express (<0.5.0,>=0.4.1)
Requires-Dist: pytest-profiling (<2.0.0,>=1.7.0)
Requires-Dist: python-dotenv (<2.0.0,>=1.0.1)
Requires-Dist: pyyaml (<7.0.0,>=6.0.1)
Requires-Dist: pyzmq (==26.0.0)
Requires-Dist: sae-vis (<0.3.0,>=0.2.18)
Requires-Dist: safetensors (<0.5.0,>=0.4.2)
Requires-Dist: transformer-lens (<2.0.0,>=1.14.0)
Requires-Dist: transformers (<5.0.0,>=4.38.1)
Requires-Dist: typer (<0.13.0,>=0.12.3)
Provides-Extra: mamba
Description-Content-Type: text/markdown
[Description omitted; length: 3380 characters]

WHEEL

Wheel-Version: 1.0
Generator: poetry-core 1.9.0
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
sae_lens/__init__.py sha256=zR3wYR3b34L-m1G6fWXefzBOtC1HyqA_SMW6tThWCIw 1098
sae_lens/analysis/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
sae_lens/analysis/dashboard_runner.py sha256=thCQnY40TVriArfx4slqxvQVXX8ZsYVNRakiTNEr8vk 15645
sae_lens/analysis/feature_statistics.py sha256=bZLSQh2WuLDWsVIofvusOubZpqKiCJTt-YKSbqTgm18 3535
sae_lens/analysis/neuronpedia_integration.py sha256=NZ9fguPyIqkjCqPaqq5ztjSQegq8LfYuLzdFmtzOO8E 17062
sae_lens/analysis/neuronpedia_runner.py sha256=jrcR-DGIQgGmYcLB9AXaKp0ei7JXlL3i3nIo6JwdXsM 20134
sae_lens/analysis/tsea.py sha256=Lbj6t4IFvj8yUq_5d8f5HGtqNlsTRrlVcVXuKUi1vg4 18275
sae_lens/pretrained_saes.yaml sha256=Prs1BT23J_-Kagtx0WmIIckcAqzDW6YJHMvLTP8EByI 1164
sae_lens/toolkit/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
sae_lens/toolkit/pretrained_sae_loaders.py sha256=zArDK14znEp6p9G78vRAkbMbhbfC7Rmjmn35CTHQxuw 2674
sae_lens/toolkit/pretrained_saes.py sha256=_OvedDOLEuRgZyHi4ruFfRD4ONX9qUC5DEDXSR1IclU 6844
sae_lens/toolkit/pretrained_saes_directory.py sha256=QXq8ykUbTuNvYvipT-aCTzSkpH83KFeopsMIZXXoc0s 1143
sae_lens/training/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
sae_lens/training/activation_functions.py sha256=EJNA4_AL7QzwhOqlO6YYJP0HsX3KwWPJnkvWWPJGqvA 471
sae_lens/training/activations_store.py sha256=jd1RkYQpncxdmZsGfQX6I4_waC-9VIKOs_H7JSdevDM 20340
sae_lens/training/batching.py sha256=buS1VBx2i2akUxxHE4WpssJgUInArFoztY4oTMIbrGM 4052
sae_lens/training/cache_activations_runner.py sha256=Mgg0DHmFpR2D2bEn9zd-_4mWZLqztiD0ot6APL7MmdY 6065
sae_lens/training/config.py sha256=UgU5Hzq4K1lh6xBV2WqzrbnIAMbJgOXCV5FuIzTYJrk 16237
sae_lens/training/evals.py sha256=6Y2bdIQY6lu4EYBHLIaJrU_GFybFTbuvBTmxe6nS33o 8229
sae_lens/training/geometric_median.py sha256=zZMzHH-tJPaLNAa48GqvzWyhpaO0VLeQj2z9pMht-9M 3747
sae_lens/training/lm_runner.py sha256=9Elrn2oy3IED757cks5u4DzUAzQhZPwU-1SIt7Tte5E 4377
sae_lens/training/load_model.py sha256=b0bB--IYce0dQxeRvhvTZ-ViplG7Hgj5di7M0upSywM 1280
sae_lens/training/optim.py sha256=y3tUYavS0nRB9Fke0jqlJQGXgsSVgsEBxHZrHXeTBYc 5574
sae_lens/training/pretokenize_runner.py sha256=x9hkDiZ0y5Rh80mJJNpO5eV3Of5L1OWFuae-UN54jAI 5401
sae_lens/training/sae_group.py sha256=ZCUYz6NsDCVW-oY1H2JnR4Q63Zcz6-KduKqdzAMvWKU 8403
sae_lens/training/session_loader.py sha256=VYZnK55_jvXqpQBbR0znGRu5QEfqe8aegmypILXZWY4 2424
sae_lens/training/sparse_autoencoder.py sha256=suquEevgxBbFwq9A_3v5HnckYak0BVP82emb6DPLaMw 20000
sae_lens/training/toy_model_runner.py sha256=SnKdU2Cj-YiPTsk_42xYf0FX5G-JmCRUYxhmMNihSlY 1932
sae_lens/training/toy_models.py sha256=7DtPgdcX_1ji1mU_IpR-vV4dI2gnQ8Zd4WOIixpgpmc 18456
sae_lens/training/train_sae_on_language_model.py sha256=mt7XHt3Tzi_zOt5EjnyHLNws_FwcrIjNmyQyugS1QE8 31427
sae_lens/training/train_sae_on_toy_model.py sha256=VwB16ldli1Xf-V0qh8qrtJaXeMtWB0CyGgYDlgNaBEI 5002
sae_lens/training/utils.py sha256=bo2OU-7IkecC1yFCccML9L6uo0CsxzSQOmKHn-WGyqw 423
sae_lens-2.1.3.dist-info/LICENSE sha256=DW6e-hDosiu4CfW0-imI57sV1I5f9UEslpviNQcOAKs 1069
sae_lens-2.1.3.dist-info/METADATA sha256=6v-tIT-8cC0n5e-ydm2nd-yIXCIHHBS8HsU1OltUIq0 5024
sae_lens-2.1.3.dist-info/WHEEL sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg 88
sae_lens-2.1.3.dist-info/RECORD