Model Based Reinforcement Learning for Atari. - Lukasz Kaiser; Mohammad Babaeizadeh; Piotr Milos; Blazej Osinski; Roy H. Campbell; Konrad Czechowski; Dumitru Erhan; Chelsea Finn; Piotr Kozakowski; Sergey Levine; Afroz Mohiuddin; Ryan Sepassi; et al. (ICLR, 2020)

Rethinking Attention with Performers. - Krzysztof Marcin Choromanski; Valerii Likhosherstov; David Dohan; Xingyou Song; Andreea Gane; Tamás Sarlós; Peter Hawkins; Jared Quincy Davis; Afroz Mohiuddin; Lukasz Kaiser; David Benjamin Belanger; Lucy J. Colwell; et al. (ICLR, 2021)

Sparse is Enough in Scaling Transformers. - Sebastian Jaszczur; Aakanksha Chowdhery; Afroz Mohiuddin; Lukasz Kaiser; Wojciech Gajewski; Henryk Michalewski; Jonni Kanerva (NeurIPS, 2021)

Q-Value Weighted Regression: Reinforcement Learning with Limited Data. - Piotr Kozakowski; Lukasz Kaiser; Henryk Michalewski; Afroz Mohiuddin; Katarzyna Kanska (IJCNN, 2022)

Scaling Up Models and Data with t5x and seqio. - Adam Roberts; Hyung Won Chung; Anselm Levskaya; Gaurav Mishra; James Bradbury; Daniel Andor; Sharan Narang; Brian Lester; Colin Gaffney; Afroz Mohiuddin; Curtis Hawthorne; Aitor Lewkowycz; et al. (CoRR, 2022)

Scaling Up Models and Data with t5x and seqio. - Adam Roberts; Hyung Won Chung; Gaurav Mishra; Anselm Levskaya; James Bradbury; Daniel Andor; Sharan Narang; Brian Lester; Colin Gaffney; Afroz Mohiuddin; Curtis Hawthorne; Aitor Lewkowycz; et al. (J. Mach. Learn. Res., 2023)