Cool Link
Forcing Flash Attention onto a TPU and Learning the Hard Way · Archer Zhang
—
archerzhang.me
Discuss
Original
Share
Sign In
👤
archerzhang.me/forcing-flash-attention-onto-a-tpu