Code

Add flex attention backend#203

Code

Open

Leymore wants to merge 4 commits intoSHI-Labs:mainSHI-Labs/NATTEN:mainfrom Leymore:flexLeymore/NATTEN:flexCopy head branch name to clipboard

+915-16

Commits

Commits on Mar 3, 2025

• add flex attention backend

  Leymore

  committed
  af4072b

  Copy full SHA for af4072b

• Integrate flex attention backend

  alihassanijr

  committed
  85fdaca

  Copy full SHA for 85fdaca

Commits on Mar 4, 2025

• Fix unit tests, formatting changes

  alihassanijr

  committed
  e12c710

  Copy full SHA for e12c710

• Update docs, remove python cache

  Leymore

  committed
  5f52f8a

  Copy full SHA for 5f52f8a