A sparse autoencoder for mechanistic interpretability research.
pip install sparse_autoencoder
Check out the demo notebook for a guide to using this library.
This project uses Poetry for dependency management, and
PoeThePoet for scripts. After checking out the repo,
we recommend setting poetry's config to create the .venv
in the root directory (note this is a
global setting) and then installing with the dev and demos dependencies.
poetry config virtualenvs.in-project true
poetry install --with dev,demos
For a full list of available commands (e.g. test
or typecheck
), run this in your terminal
(assumes the venv is active already).
poe