GitHub repository: tomoyukun/Pytorch-Correlation-extension
Description: Custom implementation of Corrleation Module