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