flash-attn-jax

PyPI page
Home page
Author: None
License: BSD-3-Clause
Summary: Flash Attention port for JAX
Latest version: 0.6.2
Required dependencies: jax | nanobind

Downloads last day: 8
Downloads last week: 55
Downloads last month: 124