/SAEs

Primary LanguageJupyter Notebook

SAEs

sae_analysis.ipynb - some exploratory analysis on GPT2-small SAEs

train.py - training code with SAE lens library