Mechanistic Interpretability visualizations.
See here for a colab demo of how to use them with the TransformerLens ActivationCache object.
You can install this library using:
pip install git+https://github.com/callummcdougall/CircuitsVis.git#subdirectory=python
Make sure to also run:
pip install transformer_lens
