Reformer, the efficient Transformer, in Pytorch - View it on GitHub