Transformers Represent Belief State Geometry in their Residual Stream - Shai, Adam S.; Marzen, Sarah E.; Teixeira, Lucas; Oldenziel, Alexander Gietelink; Riechers, Paul M. (Advances in Neural Information Processing Systems, 2024)