sae-lens

View on PyPIReverse Dependencies (3)

5.3.0 sae_lens-5.3.0-py3-none-any.whl

Wheel Details

Project: sae-lens
Version: 5.3.0
Filename: sae_lens-5.3.0-py3-none-any.whl
Download: [link]
Size: 143187
MD5: 34bbd43505625ef2344828f80122bfb9
SHA256: 5cd9df8cf63baef6dc4c093728a8359a26ad3d7c3549f15f068cd39a4eb8f441
Uploaded: 2024-12-29 16:53:59 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: sae-lens
Version: 5.3.0
Summary: Training and Analyzing Sparse Autoencoders (SAEs)
Author: Joseph Bloom
Home-Page: https://jbloomaus.github.io/SAELens
Project-Url: Repository, https://github.com/jbloomAus/SAELens
License: MIT
Keywords: deep-learning,sparse-autoencoders,mechanistic-interpretability,PyTorch
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.10,<4.0
Requires-Dist: automated-interpretability (<1.0.0,>=0.0.5)
Requires-Dist: babe (<0.0.8,>=0.0.7)
Requires-Dist: datasets (<3.0.0,>=2.17.1)
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: 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: safetensors (<0.5.0,>=0.4.2)
Requires-Dist: simple-parsing (<0.2.0,>=0.1.6)
Requires-Dist: transformer-lens (<3.0.0,>=2.0.0)
Requires-Dist: transformers (<5.0.0,>=4.38.1)
Requires-Dist: typer (<0.13.0,>=0.12.3)
Requires-Dist: typing-extensions (<5.0.0,>=4.10.0)
Requires-Dist: zstandard (<0.23.0,>=0.22.0)
Provides-Extra: mamba
Description-Content-Type: text/markdown
[Description omitted; length: 3491 characters]

WHEEL

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

RECORD

Path Digest Size
sae_lens/__init__.py sha256=jLNJBnVHZpWAJhLgaQFtguzlBCoT97ZaB7qVYMmvVEk 1124
sae_lens/analysis/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
sae_lens/analysis/feature_statistics.py sha256=5iIhjBqtCDEG9C6ZMncc672FCC5X5TIOE0sCMkEwIP4 3210
sae_lens/analysis/hooked_sae_transformer.py sha256=uJpmLNm6dnIFWvC3WE4EPTuk5qChbmh7WvYQOtzswdU 13976
sae_lens/analysis/neuronpedia_integration.py sha256=KB_PRTvHHgTmPfBphVvbOqrXB_bFQ986lwrETGgySYU 18501
sae_lens/analysis/tsea.py sha256=B9wIlzRjWMvQvu-SoruORzTMKn084QHa5tdYKqiYYD8 18179
sae_lens/cache_activations_runner.py sha256=FkN6k1Ay_GSzewTCnAIrqfStmQbtshH4kJWCmWqnV7o 11735
sae_lens/config.py sha256=e5ox6Uii959w-EhNvXXNnDu-1E0Ys-2YU5QYxkPNXNQ 33912
sae_lens/evals.py sha256=Us5svQerjiSSCNkXxxjpv953n0fgFTpjGqIeGKF4PAQ 36937
sae_lens/load_model.py sha256=tE70sXsyyyGYW7o506O3eiw1MXyyW6DCQojLG49hWYI 6771
sae_lens/pretokenize_runner.py sha256=wHBwajIh-w2G07mguFN8gkLsdp278C9KFCU6BWvI_38 6195
sae_lens/pretrained_saes.yaml sha256=AZ3qHrhGWZKN6XYhwE1a8EwttvTtDpshMTBPF1xAwUA 603367
sae_lens/sae.py sha256=ms12ujjO5rexvVO-75_eQ6t83qBXQ6qoZgZ3zSlAEdY 27281
sae_lens/sae_training_runner.py sha256=VF-FQLwU4iBI7Q1yhFUc955kW9RzQ_BXPYsLogERu_U 8843
sae_lens/tokenization_and_batching.py sha256=oUAscjy_LPOrOb8_Ty6eLAcZ0B3HB_wiWjWktgolhG0 4314
sae_lens/toolkit/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
sae_lens/toolkit/pretrained_sae_loaders.py sha256=n7rdB_7uzIH3bGO3lnnWOj18GcoIVOaU6Juu9eo9ULE 23128
sae_lens/toolkit/pretrained_saes.py sha256=cpQO7-WGqdW1I2ODNWy1GyNTTdt_pl8pny5-YKn8DBk 4800
sae_lens/toolkit/pretrained_saes_directory.py sha256=H8FSek-mUL2O-oQcv_U9jcT-5_ncwWNKJKBfeCScQxo 3273
sae_lens/toy_model_runner.py sha256=M1Lj2L9mItDX86uP4jqckxdQ6jL8c5DZS6jPKVqnTn8 1902
sae_lens/training/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
sae_lens/training/activations_store.py sha256=HjIwMt2MYGpv2tK3QxI7a7OOCT4WRoXlZ_qO3g3G6cY 31365
sae_lens/training/geometric_median.py sha256=_Koprzj9nyB_G7h_4YENePohMTZVYLtyKhJx6sU_lKw 3750
sae_lens/training/optim.py sha256=AImcc-MAaGDLOBP2hJ4alDFCtaqqgm4cc2eBxIxiQAo 5784
sae_lens/training/sae_trainer.py sha256=YZFyKCawfkXroBVlSolPoghBQJjrjOZnHjm2wVAIONA 16357
sae_lens/training/toy_models.py sha256=339fF8QP-QxaE3U50M1u4wDwbGXNzcy7eKkpBoP8oJQ 18419
sae_lens/training/train_toy_sae.py sha256=F6B-5NYRE6JbMsPHY6gM7uFd8rNAo575_lpC2lFzBZM 4916
sae_lens/training/training_sae.py sha256=MR0qfoq8CGgWwaXku_FytCaDLXp5Q5xcsL0df9tBn7E 27476
sae_lens/training/upload_saes_to_huggingface.py sha256=xcB1FffAKdG8XyqjLIm-FPP_i5db3ay-kfXUSUy-zx8 4317
sae_lens-5.3.0.dist-info/LICENSE sha256=DW6e-hDosiu4CfW0-imI57sV1I5f9UEslpviNQcOAKs 1069
sae_lens-5.3.0.dist-info/METADATA sha256=ArS7T7rG-usS_RAnOhW2Yhz30Mc1SRxGYP_cNHcc9MY 5198
sae_lens-5.3.0.dist-info/WHEEL sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs 88
sae_lens-5.3.0.dist-info/RECORD