PyPI page
Home page
Author:
Xuyang Ge, Zhengfu He, Wentao Shu, Fukang Zhu, Lingjie Chen, Junxuan Wang
License:
MIT
Summary:
For OpenMOSS Mechanistic Interpretability Team's Sparse Autoencoder (SAE) research. Open-sourced and constantly updated.
Latest version:
2.0.0b34
Required dependencies:
cattrs
|
datasets
|
einops
|
jaxtyping
|
json-repair
|
more-itertools
|
numpy
|
openai
|
pandas
|
pydantic
|
pydantic-settings
|
pymongo
|
pyyaml
|
rich
|
safetensors
|
tomlkit
|
torch
|
transformer-lens
|
transformers
|
typer
|
typing-extensions
|
wandb
Optional dependencies:
sae-lens
|
torch-npu
|
triton
Downloads last day:
10
Downloads last week:
112
Downloads last month:
365