andyrdt/saes-llama-3.1-8b-instruct

Residual stream SAEs for Llama-3.1-8B-Instruct.

These SAEs were trained using a blend of chat (lmsys/lmsys-chat-1m) and pretraining data (monology/pile-uncopyrighted), and also a small amount of emergent misalignment data.

Each SAE is trained using BatchTopK. For each layer, we train 4 SAEs, with k=32,64,128,256.

For more training details, see https://github.com/andyrdt/dictionary_learning/tree/andyrdt/llama_saes.