gitaskhub

Get down and dirty with FlashAttention2.0 in pytorch, plug in and play no complex CUDA kernels

Stars · 115
Language · Python
License · MIT
Ask anything about this repo to start.

By chatting or signing in you agree to the Terms and chat-message logging (revocable in History).