Try

Enter a prompt and run a forward pass to watch all 144 attention heads at once.