A flexible and efficient implementation of Flash Attention 2.0 for JAX, supporting multiple backends (GPU/TPU/CPU) and platforms (Triton/Pallas/JAX).
-
Updated
Mar 4, 2025 - Python
#
pallas
Here are 8 public repositories matching this topic...
Calculate the hash of any input for ZK-Friendly hashes (MiMC & Poseidon) over a variety of Elliptic Curves.
-
Updated
Feb 7, 2023 - TypeScript
A library used to access the Pallas service
-
Updated
Feb 14, 2025 - Python
Benchmarking the JAX Pallas implementation of a custom RNN against alternatives
-
Updated
Feb 27, 2025 - Python
Repo to hold core components when building a Pallas Systems Website
-
Updated
Mar 10, 2025 - TypeScript
Improve this page
Add a description, image, and links to the pallas topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the pallas topic, visit your repo's landing page and select "manage topics."