Comparing 5 ways to implement Multihead Attention in PyTorch
github.com