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}},
}
Release files for sae-lens 6.54.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| sae_lens-6.54.0.tar.gz | 287.0 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| sae_lens-6.54.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 607.9 kB
Release files / sae_lens-6.54.0.tar.gz
| Download URL | sae_lens-6.54.0.tar.gz |
|---|---|
| Size | 287.0 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
4527cfe8c23d0822ddd2ce7e99f48213debf76769940e7ddfc77409cb78fac46
|
|
BLAKE2b-256 checksum How to use checksums |
ce5c34db356d9727db0da2beb41cebfab38868f66c51ad6982f697243464f81a
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Oct 4, 2026.
Transparency logRelease files / sae_lens-6.54.0-py3-none-any.whl
| Download URL | sae_lens-6.54.0-py3-none-any.whl |
|---|---|
| Size | 320.9 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
47398334573a9e2fc427d6e79c7df24bdb8ded728e9c1c81a60df1f180af0440
|
|
BLAKE2b-256 checksum How to use checksums |
a05f828d2d3774a88ea7e57876f5592b456db0887cd2d149aa5ba76e778cb7df
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Oct 4, 2026.
Transparency log
