Get our free extension to see links to code for papers anywhere online!

Chrome logo Add to Chrome

Firefox logo Add to Firefox

Picture for Anselm Levskaya

Exploring the limits of Concurrency in ML Training on Google TPUs


Nov 07, 2020
Sameer Kumar, James Bradbury, Cliff Young, Yu Emma Wang, Anselm Levskaya, Blake Hechtman, Dehao Chen, HyoukJoong Lee, Mehmet Deveci, Naveen Kumar, Pankaj Kanwar, Shibo Wang, Skye Wanderman-Milne, Steve Lacy, Tao Wang, Tayo Oguntebi, Yazhou Zu, Yuanzhong Xu, Andy Swing


  Access Paper or Ask Questions

Reformer: The Efficient Transformer


Feb 18, 2020
Nikita Kitaev, Łukasz Kaiser, Anselm Levskaya

* ICLR 2020 

  Access Paper or Ask Questions

Stand-Alone Self-Attention in Vision Models


Jun 13, 2019
Prajit Ramachandran, Niki Parmar, Ashish Vaswani, Irwan Bello, Anselm Levskaya, Jonathon Shlens


  Access Paper or Ask Questions