In-Context Learning with Representations: Contextual Generalization of Trained Transformers Tong Y ang CMU Y u Huang