When XLA Isn't Enough: From Pallas to VLIW with Splash Attention on TPU

1 points | by patrick_toulme 15 hours ago