dogear

enter for all results · esc to close

kvax

github.comtool

A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.

visit github.com ↗
0
sign in to dogear this
from
JAX
added
2026-10-10
likes
0

JAX › Libraries > New Libraries: “A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.”