Github Self Attention Example