or
This repo is a reimplementation for RASP as really neat language for thinking about transformer calculations.
This version is done completely in Python as an embedded domain specific language.