SAE Lens
SAELens exists to help researchers:
- Train sparse autoencoders.
- Analyse sparse autoencoders / research mechanistic interpretability.
- Generate insights which make it easier to create safe and aligned AI systems.
SAELens inference works with any PyTorch-based model, not just TransformerLens. While we provide deep integration with TransformerLens via HookedSAETransformer, SAEs can be used with Hugging Face Transformers, NNsight, or any other framework by extracting activations and passing them to the SAE's encode() and decode() methods.
Please refer to the documentation for information on how to:
- Download and Analyse pre-trained sparse autoencoders.
- Train your own sparse autoencoders.
- Generate feature dashboards with the SAE-Vis Library.
SAE Lens is the result of many contributors working collectively to improve humanity's understanding of neural networks, many of whom are motivated by a desire to safeguard humanity from risks posed by artificial intelligence.
This library is maintained by Joseph Bloom, Curt Tigges, Anthony Duong and David Chanin.
Loading Pre-trained SAEs.
Pre-trained SAEs for various models can be imported via SAE Lens. See this page for a list of all SAEs.
Migrating to SAELens v6
The new v6 update is a major refactor to SAELens and changes the way training code is structured. Check out the migration guide for more details.
Tutorials
- SAE Lens + Neuronpedia
- Loading and Analysing Pre-Trained Sparse Autoencoders
- Understanding SAE Features with the Logit Lens
- Training a Sparse Autoencoder
- Training SAEs on Synthetic Data
- SynthSAEBench: Evaluating SAE Architectures on Synthetic Data
Join the Slack!
Feel free to join the Open Source Mechanistic Interpretability Slack for support!
Other SAE Projects
- dictionary-learning: An SAE training library that focuses on having hackable code.
- Sparsify: A lean SAE training library focused on TopK SAEs.
- Overcomplete: SAE training library focused on vision models.
- SAE-Vis: A library for visualizing SAE features, works with SAELens.
- SAEBench: A suite of LLM SAE benchmarks, works with SAELens.
Citation
Please cite the package as follows:
@misc{bloom2024saetrainingcodebase,
title = {SAELens},
author = {Bloom, Joseph and Tigges, Curt and Duong, Anthony and Chanin, David},
year = {2024},
howpublished = {\url{https://github.com/decoderesearch/SAELens}},
}
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file sae_lens-6.49.2.tar.gz.
File metadata
- Download URL: sae_lens-6.49.2.tar.gz
- Upload date:
- Size: 276.8 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via:
twine/7.0.0 CPython/3.13.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
6831935ca839ce6cec658fdac3a7d98b646ed2cf4cebbf53c8871084b6453868
|
|
| MD5 |
0944644a657a4f94765b871afa3c283a
|
|
| BLAKE2b-256 |
af7f58e82a260d386c09636eb53e37d9609cf889ca977d6df4e035d35fbdc2e7
|
Provenance
The following attestation bundles were made for sae_lens-6.49.2.tar.gz:
Publisher:
build.yml on decoderesearch/SAELens
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
sae_lens-6.49.2.tar.gz -
Subject digest:
6831935ca839ce6cec658fdac3a7d98b646ed2cf4cebbf53c8871084b6453868 - Sigstore transparency entry: 2647755001
- Sigstore integration time:
-
Permalink:
decoderesearch/SAELens@48c0ea92aac5b3684d2c4cafd8b8dc3c19c9349a -
Branch / Tag:
refs/heads/main - Owner: https://github.com/decoderesearch
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
build.yml@48c0ea92aac5b3684d2c4cafd8b8dc3c19c9349a -
Trigger Event:
push
-
Statement type:
File details
Details for the file sae_lens-6.49.2-py3-none-any.whl.
File metadata
- Download URL: sae_lens-6.49.2-py3-none-any.whl
- Upload date:
- Size: 313.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via:
twine/7.0.0 CPython/3.13.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
7e8c9aa6c34fc5eaecbc51ceed4e60db4db6b12d21859acfba1e1ccd1ca53510
|
|
| MD5 |
993b0963dbe3d78b23df30242959507f
|
|
| BLAKE2b-256 |
721f966f4ac2f147a2199fbaa04eeb95ef53ff4c85dc0061cb138fb002d218cd
|
Provenance
The following attestation bundles were made for sae_lens-6.49.2-py3-none-any.whl:
Publisher:
build.yml on decoderesearch/SAELens
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
sae_lens-6.49.2-py3-none-any.whl -
Subject digest:
7e8c9aa6c34fc5eaecbc51ceed4e60db4db6b12d21859acfba1e1ccd1ca53510 - Sigstore transparency entry: 2647755017
- Sigstore integration time:
-
Permalink:
decoderesearch/SAELens@48c0ea92aac5b3684d2c4cafd8b8dc3c19c9349a -
Branch / Tag:
refs/heads/main - Owner: https://github.com/decoderesearch
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
build.yml@48c0ea92aac5b3684d2c4cafd8b8dc3c19c9349a -
Trigger Event:
push
-
Statement type:
