issues
search
nshepperd
/
flash_attn_jax
JAX bindings for Flash Attention v2
BSD 3-Clause "New" or "Revised" License
61
stars
0
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Encounter edge_padding_low error while testing
#7
sh0416
closed
1 month ago
5
Supporting varlen?
#6
sh0416
opened
1 month ago
0
performance benchmark
#5
cliangyu
opened
1 month ago
1
it will support group attention?
#4
pass-lin
closed
1 month ago
2
High difference between values from vanilla attention and flash_mha
#3
VachanVY
closed
3 months ago
8
ModuleNotFoundError: No module named `flash_attn_jax.flash_api`
#2
VachanVY
closed
3 months ago
6
how to install this lib by pip
#1
pass-lin
closed
3 months ago
1