Amazing! We found your original library super useful for our Attention SAEs research, so thanks for making this!
Connor Kissane
Karma: 176
We Inspected Every Head In GPT-2 Small using SAEs So You Don’t Have To
Attention SAEs Scale to GPT-2 Small
Sparse Autoencoders Work on Attention Layer Outputs
These puzzles are great, thanks for making them!
Code for this token filtering can be found in the appendix and the exact token list is linked.
Maybe I just missed it, but I’m not seeing this. Is the code still available?
Thanks for the comment! We always use the pre-ReLU feature activation, which is equal to the post-ReLU activation (given that the feature is activate), and is purely linear function of z. Edited the post for clarity.