sae-lens

View on PyPIReverse Dependencies (1)

1.3.0 sae_lens-1.3.0-py3-none-any.whl

Wheel Details

Project: sae-lens
Version: 1.3.0
Filename: sae_lens-1.3.0-py3-none-any.whl
Download: [link]
Size: 65679
MD5: 6bb12f3ce0b4332e43e61239b0150b60
SHA256: 0e47d169f2e1a17578605a8ac225dde1b695cc1c797aa4b59b9fa898ca02b987
Uploaded: 2024-05-03 10:11:19 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: sae-lens
Version: 1.3.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: 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: 2552 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=1xDDYsC06Ai-2-2kHMgELB0wnOrn4CMkyYluDCt_XWI 943
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=5_L-nWIPbovNUv78p7dshF_XO2TDUAGM3gZ4kwQtDwM 6643
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=6kP1HJNF0zc7CxZbInH8g3kPL6qsbuHcWJ13fpuOyAY 16793
sae_lens/training/cache_activations_runner.py sha256=P5NaVaZ0vGGc7M0dNnsFf-GVTyUfAROLIleZqUf6Q7Y 2849
sae_lens/training/config.py sha256=zFRnsS9X7ST3CJkd86t_U57dVH9A2mFYoy225X4KtmE 10112
sae_lens/training/evals.py sha256=zAykS8XLviTWlSXcrtxAXkQ9GL5JL6gFxn8KXan6dv4 6169
sae_lens/training/geometric_median.py sha256=zZMzHH-tJPaLNAa48GqvzWyhpaO0VLeQj2z9pMht-9M 3747
sae_lens/training/lm_runner.py sha256=Tmp_VQZMFdfYtupMFJTr4f1MCASfn7fHbTqllfURtXo 1528
sae_lens/training/load_model.py sha256=g_ZrG9oySu6cPDN4RWVjQ5cqAI5frDl5AXuz21SXdJo 1020
sae_lens/training/optim.py sha256=0sUXOFgxOv4pzLfkL5haTOMqpnt8hSXKI9fnuuqUVNw 3675
sae_lens/training/sae_group.py sha256=mlSn0cmpOOv8Do3ZDxBG9TZ1sXTiwuE1zljcnOI0kMI 8266
sae_lens/training/session_loader.py sha256=ZG9R-fyG8HiE1k2GrEy2Lg886Q2k1vUYBeRpkA2a9sI 2319
sae_lens/training/sparse_autoencoder.py sha256=DnSVme2C_h45Cx9xHyMRZgrj0oKug0pZh8Ie2ojjJaY 17403
sae_lens/training/toy_model_runner.py sha256=BhoDkioz_ZXxQOnRtYPMPk9i3zaoexH8uTeJHFdB_b0 3276
sae_lens/training/toy_models.py sha256=XM7gDxPCUMdWob2qre-cAd0nB5AFr_PTachPkd8hONw 17362
sae_lens/training/train_sae_on_language_model.py sha256=rxL9hWcdXmjVBfcV2g0WM7trzVI7As8cKWKi01xymWU 19601
sae_lens/training/train_sae_on_toy_model.py sha256=VwB16ldli1Xf-V0qh8qrtJaXeMtWB0CyGgYDlgNaBEI 5002
sae_lens/training/utils.py sha256=-4l_uM6ZvbWnF6prm49x4EgXlL9Nn7zzTS0pMl_meZo 1628
sae_lens-1.3.0.dist-info/LICENSE sha256=DW6e-hDosiu4CfW0-imI57sV1I5f9UEslpviNQcOAKs 1069
sae_lens-1.3.0.dist-info/METADATA sha256=Sfe5famkG55s0MS3wdNxapyzppWToHJiyAfIC_8iBVg 4147
sae_lens-1.3.0.dist-info/WHEEL sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg 88
sae_lens-1.3.0.dist-info/RECORD