Implementation of Poly-attention, a higher-order self-attention proposed by Chakrabarti et al. of Columbia - View it on GitHub