Profiling in PyTorch (Part 3): Attention is all you profile