https://pypi.org/project/fa4-jax/ Recent updates to the Python Package Index for fa4-jax en https://pypi.org/project/fa4-jax/0.1.0/ FlashAttention-4 (CuTeDSL) for JAX: fwd/bwd/jvp with flex-attention mask mods, via cutejax nshepperd@gmail.com Sun, 11 Oct 2026 13:14:35 GMT