2 ms·When XLA Isn't Enough: From Pallas to VLIW with Splash Attention on TPU1 points by matt_d 10mo ago