Code for High-dimensional layer-state trajectories reveal the internal dynamics of transformer inference

Part of

Code for High-dimensional layer-state trajectories reveal the internal