Releases: lucidrains/flash-attention-jax
Releases · lucidrains/flash-attention-jax
0.3.1
0.3.0
make sure causal flash attention can have batch and head dimensions
0.2.0
0.2.0
0.1.0
release 0.1.0
0.0.10
add batch and multihead support for non-causal flash attention, for @…
0.0.9
fix cosine sim attention precision
0.0.8
fix for cosine sim attention
0.0.7
offer cosine sim flash attention variant
0.0.6
0.0.6
v0.0.5
make sure causal flash attention works even if key and query lengths …