A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude. - View it on GitHub