transformer-lens

An implementation of transformers tailored for mechanistic interpretability.


License
MIT
Install
pip install transformer-lens==2.4.1