These have much more mature autoencoders: https://github.com/jbloomAus/SAELens?tab=readme-ov-file Much more time efficient to just use these rather than reimplement from scratch.