Official PyTorch implementation for Implicit Kernel Attention [AAAI 2021]
Primary LanguagePythonApache License 2.0Apache-2.0