Forcing Flash Attention onto a TPU and Learning the Hard Wayarcherzhang.me58 points·by azhng·vor 4 Monaten·16 comments