nshepperd/flash_attn_jax

Issues

• Encounter edge_padding_low error while testing

  #7 opened 2 months ago
  5

• Supporting varlen?

  #6 opened 2 months ago
  0

• performance benchmark

  #5 opened 2 months ago
  1

• it will support group attention?

  #4 opened 2 months ago
  2

• High difference between values from vanilla attention and flash_mha

  #3 opened 4 months ago
  8

• ModuleNotFoundError: No module named `flash_attn_jax.flash_api`

  #2 opened 4 months ago
  6

• how to install this lib by pip

  #1 opened 4 months ago
  1