Flash Attention in ~100 lines of CUDA (forward pass only)
By chatting or signing in you agree to the Terms and chat-message logging (revocable in History).