SAEs sae_analysis.ipynb - some exploratory analysis on GPT2-small SAEs train.py - training code with SAE lens library