az

icon indicating a website link alexzhang13.github.io icon indicating an email [email protected]

icon location NYC icon location @aivant | Incoming PhD student at MIT, prev. Princeton CS

Results 1 repositories owned by az

flashattention2-custom-mask

65
Stars
5
Forks
Watchers

Triton implementation of FlashAttention2 that adds Custom Masks.

verified publisher icon alexzhang13