Transformers Can Do Arithmetic with the Right Embeddings - McLeish, Sean; Bansal, Arpit; Stein, Alex; Jain, Neel; Kirchenbauer, John; Bartoldson, Brian R.; Kailkhura, Bhavya; Bhatele, Abhinav; Geiping, Jonas; Schwarzschild, Avi; Goldstein, Tom (Advances in Neural Information Processing Systems, 2024)