Adam Shai comments on Transformers Represent Belief State Geometry in their Residual Stream