Implementation of the proposed DeepCrossAttention by Heddes et al at Google research, in Pytorch - View it on GitHub