flash-attention-jax

PyPI page
Home page
Author: Phil Wang
License: MIT
Summary: Flash Attention - in Jax
Latest version: 0.3.1
Required dependencies: einops | jax

Downloads last day: 0
Downloads last week: 15
Downloads last month: 79