A flexible and efficient implementation of Flash Attention 2.0 for JAX, supporting multiple backends (GPU/TPU/CPU) and platforms (Triton/Pallas/JAX).

A FlashAttention backwards-over-backwards ⚡🔙🔙

Calculate the hash of any input for ZK-Friendly hashes (MiMC & Poseidon) over a variety of Elliptic Curves.

pdf files for alchi https://github.com/milahu/alchi

Variance-stable routing for 2-bit quantized MoE models. Features dynamic phase correction (Armen Guard), syntactic stabilization layer, and recursive residual quantization for efficient inference.

Repo to hold core components when building a Pallas Systems Website

Benchmarking the JAX Pallas implementation of a custom RNN against alternatives

Main Website for Pallas Systems

Public Domain Works of Peter Simon Pallas (1741 - 1811)

A fork of Pallas-Puzzles aimed at migrating to the modern pallas-way of doing things.

SuperNova (Pasta) proof generator & verifier with CI and frozen fixtures