amudide/switch_sae

Efficient Dictionary Learning with Switch Sparse Autoencoders (SAEs)

Jupyter Notebook

More soon!

Credits

This repository is adapted from dictionary_learning by Samuel Marks and Aaron Mueller.