Efficiently Scaling Transformer Inference. - Reiner Pope; Sholto Douglas; Aakanksha Chowdhery; Jacob Devlin; James Bradbury; Anselm Levskaya; Jonathan Heek; Kefan Xiao; Shivani Agrawal; Jeff Dean (CoRR, 2022)

Efficiently Scaling Transformer Inference. - Reiner Pope; Sholto Douglas; Aakanksha Chowdhery; Jacob Devlin; James Bradbury; Jonathan Heek; Kefan Xiao; Shivani Agrawal; Jeff Dean (MLSys, 2023)

Training Chain-of-Thought via Latent-Variable Inference. - Matthew Douglas Hoffman; Du Phan; David Dohan; Sholto Douglas; Tuan Anh Le; Aaron Parisi; Pavel Sountsov; Charles Sutton; Sharad Vikram; Rif A. Saurous (NeurIPS, 2023)