Confident Adaptive Language Modeling - Schuster, Tal; Fisch, Adam; Gupta, Jai; Dehghani, Mostafa; Bahri, Dara; Tran, Vinh; Tay, Yi; Metzler, Donald (Advances in Neural Information Processing Systems, 2022)