Self-Attention does not need O(n^2) Memory
arxiv.org