Transformers Represent Belief State Geometry in their Residual Stream Adam S. Shai Sarah E. Marzen Lucas Teixeira Simplex Department of Natural Sciences PIBBSS