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: 6
Downloads last week: 25
Downloads last month: 102