A collection of some attention / convolution operators implemented using PyTorch.
Primary LanguagePythonMIT LicenseMIT