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: 2
Downloads last week: 38
Downloads last month: 544