flash-attn3-jax

PyPI page
Home page
Author: None
License: BSD-3-Clause
Summary: FlashAttention-3 JAX binding using XLA FFI
Latest version: 0.2.3
Required dependencies: einops | jax

Downloads last day: 0
Downloads last week: 17
Downloads last month: 149