Metadata-Version: 2.1
Name: sparse_autoencoder
Version: 0.3.0
Summary: Sparse Autoencoder for Mechanistic Interpretability
License: MIT
Author: Alan Cooney
Author-email: 41682961+alan-cooney@users.noreply.github.com
Requires-Python: >=3.10,<3.13
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Requires-Dist: einops (>=0.6)
Requires-Dist: torch (>=2.1)
Requires-Dist: wandb (>=0.15.12)
Description-Content-Type: text/markdown

# Sparse Autoencoder

[![Pypi](https://img.shields.io/pypi/v/sparse_autoencoder?color=blue)](https://pypi.org/project/transformer-lens/)
![PyPI -
License](https://img.shields.io/pypi/l/sparse_autoencoder?color=blue) [![Checks](https://github.com/alan-cooney/sparse_autoencoder/actions/workflows/checks.yml/badge.svg)](https://github.com/alan-cooney/sparse_autoencoder/actions/workflows/checks.yml)
[![Release](https://github.com/alan-cooney/sparse_autoencoder/actions/workflows/release.yml/badge.svg)](https://github.com/alan-cooney/sparse_autoencoder/actions/workflows/release.yml)

A sparse autoencoder for mechanistic interpretability research.


```shell
pip install sparse_autoencoder
```

## Demo

Check out the demo notebook for a guide to using this library.

## Contributing

This project uses [Poetry](https://python-poetry.org) for dependency management, and
[PoeThePoet](https://poethepoet.natn.io/installation.html) for scripts. After checking out
the repo, install all dependencies with:

```shell
poetry install --with dev,demos
```

Then for a full list of available commands (e.g. `test` or `typecheck`):

```shell
poe
```

