#

splash-attention

Here is 1 public repository matching this topic...

ashioyajotham / recompute_dont_restore

Flash Attention from first principles on TPU using JAX Pallas.

pallas tpu jax flash-attention splash-attention

Updated Apr 28, 2026
Python

Improve this page

Add a description, image, and links to the splash-attention topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the splash-attention topic, visit your repo's landing page and select "manage topics."