Run pip install -e .. I have written this transformer by following and solving the ARENA exercises. To test the layers, I have used the gpt from HookedTransformer.
Train the sample transformer using 1-train_sample_transformer.ipynb.
- explore transformer lens
- visualize attention patterns
- write basic detectors
- understand induction heads
- direct logit attribution
- induction head ablation