Tap Mobile interview question

During technical interview: implement MultiHeadAttention layer using PyTorch.