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: 2
Downloads last week: 7
Downloads last month: 68
7-day smoothing