Department of Electrical Engineering and Computer Sciences, University of California, Berkeley, Berkeley, CA 94709, USA; Google Research, Mountain View, CA 94043, USA
Reese Pathak, Rajat Sen, Weihao Kong, Abhimanyu Das
transformers;mixture models;linear regression
Transformers can optimally learn regression mixture models