Flash Attention & friends in pure Julia - View it on GitHub