sae-lens

View on PyPIReverse Dependencies (1)

3.2.0 sae_lens-3.2.0-py3-none-any.whl

Wheel Details

Project: sae-lens
Version: 3.2.0
Filename: sae_lens-3.2.0-py3-none-any.whl
Download: [link]
Size: 78282
MD5: 24ae352da64ae600fc02f4c80f08a493
SHA256: 64ae2381e5298d244dce38210fea5bb71d63e6c2f509e8a035bceaa97a50278d
Uploaded: 2024-05-30 14:51:41 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: sae-lens
Version: 3.2.0
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=IgEp6OPLL_9mOjjuKqC7Q7LEcWF_9ZPesFkgzzB4Hcg 898
sae_lens/analysis/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
sae_lens/analysis/feature_statistics.py sha256=nY3NoT1xW34gP4z19_ssL34jIxeOeEzzodBtEGrRIiE 3320
sae_lens/analysis/hooked_sae_transformer.py sha256=r5A6KtIrVXKQrVwQpI_Lur-ugJtxho8QWeHVmBTCfm4 12561
sae_lens/analysis/neuronpedia_integration.py sha256=NZ9fguPyIqkjCqPaqq5ztjSQegq8LfYuLzdFmtzOO8E 17062
sae_lens/analysis/neuronpedia_runner.py sha256=inM3TmuR2V3P9BaAG8fCDVikB8yApv3-Xr5pvWKvpDE 20406
sae_lens/analysis/tsea.py sha256=Lbj6t4IFvj8yUq_5d8f5HGtqNlsTRrlVcVXuKUi1vg4 18275
sae_lens/batching.py sha256=buS1VBx2i2akUxxHE4WpssJgUInArFoztY4oTMIbrGM 4052
sae_lens/cache_activations_runner.py sha256=0LnRwg3PBw04HLZBevHFfOIO7f9xit3Qr88kE_dvQww 6047
sae_lens/config.py sha256=3XyMahtJUvjUf93BSx1REx3bjU175FdbsUZiBAo-8V0 24486
sae_lens/dashboard_runner.py sha256=JducOUgIRbr4mN1yfoFg4J5DvTRglRN1xWjakg5Lf5A 16161
sae_lens/evals.py sha256=3HjUWAf2yWVKHpzLa0O46RSnZQfIByyUtE05P4eThhI 7966
sae_lens/load_model.py sha256=rHKNx9GFkPzblyQo0vWot5UUKc9ys6Ssg_2ffdvdW7k 1320
sae_lens/pretokenize_runner.py sha256=_s35HkKFGk3g6_dQz9TyGZdMp4Qd7r2_UnBj9NN1xVs 5383
sae_lens/pretrained_saes.yaml sha256=TOQfCJsqA97MOiVJ6tmTYZVWmXF8Aknu8f8Qe7_YiJ8 4833
sae_lens/sae.py sha256=TLw-aoh-0hy-lOcSDh0RxrmwLLqP9c5eGY7zYjTCfmY 14329
sae_lens/sae_training_runner.py sha256=Pvd6etQ6pQNkgKA70cBGNhwhZYZr7fgj8bQcx91sTPM 6972
sae_lens/toolkit/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
sae_lens/toolkit/pretrained_sae_loaders.py sha256=CEU8f4xzPyIQv_5FH2WcSOqdoa8umuELnIGh57oFHRQ 8087
sae_lens/toolkit/pretrained_saes.py sha256=xNNkGOxjwzCL4FcSSFTIfYyhFN0A5FvY7ZRJQddN9OM 4752
sae_lens/toolkit/pretrained_saes_directory.py sha256=QXq8ykUbTuNvYvipT-aCTzSkpH83KFeopsMIZXXoc0s 1143
sae_lens/toy_model_runner.py sha256=W7mKC1GbHO6xsnhpDE3pyMWU-5YiyzpQbw8WyZ8xZFM 1783
sae_lens/training/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
sae_lens/training/activations_store.py sha256=4UPlmbefL9-jt2lvss8cU3bL9pOMNpZX4e1No5MWST4 21162
sae_lens/training/geometric_median.py sha256=cK5vAzV_wBuNaz2JpeuPJ-0KObZZAzaQ7_80hAD05CA 3779
sae_lens/training/optim.py sha256=aBrV8QyfASL9SeqMPyZd_vA_Xm4l9VmW4l_6l3EELF4 5585
sae_lens/training/sae_trainer.py sha256=hF1F609IexSrEADS3Z9jGLigDn0e3zKbwzvABKeSMK0 14209
sae_lens/training/toy_models.py sha256=7DtPgdcX_1ji1mU_IpR-vV4dI2gnQ8Zd4WOIixpgpmc 18456
sae_lens/training/train_toy_sae.py sha256=5DXUhAATIZNAwKy9pj8NRpd_-eAN1fbfDHl4VW9sp78 4850
sae_lens/training/training_sae.py sha256=vrDLPA0gjvOcknG4R2ToDm6DaqqyhRJwErWmOp0IO4k 15495
sae_lens-3.2.0.dist-info/LICENSE sha256=DW6e-hDosiu4CfW0-imI57sV1I5f9UEslpviNQcOAKs 1069
sae_lens-3.2.0.dist-info/METADATA sha256=rhrqiSvqPgXotqp9Bw3lLnaMeOTwdu9mRmQLhEDyaeU 5024
sae_lens-3.2.0.dist-info/WHEEL sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg 88
sae_lens-3.2.0.dist-info/RECORD