GitHub repository: zzd1992/Image-Local-Attention
Primary language: Python
Description: A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.