Here's the research diary which has all the information about this project.
The main file is itsa_me_mainio.py.
The autoencoders dir is stolen borrowed from Hoagy and Aidan's SAE paper repo.
The models dir contains the SAEs I'm working with (which I got from Hoagy).