Self-attention Does Not Need O(n^2) Memory
arxiv.org