Click on Code Walkthrough: Multihead Attention to open the resource.