Add `flex_attention` backend

cellarium-ai / cellarium-ml

Distributed single-cell data analysis.

BSD 3-Clause "New" or "Revised" License

22 stars 3 forks source link

Closed ordabayevy closed 5 days ago

ordabayevy commented 2 weeks ago

Removed flash backend because it doesn't support attention mask.