TransformerCircuits. A pedagogical implementation of "A Mathematical Framework for Transformer Circuits". https://transformer-circuits.pub/2021/framework/index.html