Skip to content

Use Splash Attention on TPU when available/possible (#566) #31

Use Splash Attention on TPU when available/possible (#566)

Use Splash Attention on TPU when available/possible (#566) #31