Merge pull request #313 from philipturner/patch-1

Metal FlashAttention
Dao-AILab · Jul 16, 2023 · 01c40da · 01c40da
2 parents 72ad03e + 4dbcaa1
commit 01c40da
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/usage.md b/usage.md
@@ -123,3 +123,5 @@ yields the fastest BERT training on cloud instances in MLPerf training 2.0 (June
 
 - [Jax](https://github.com/google/jax): an [implementation](https://github.com/lucidrains/flash-attention-jax)
   in Jax by [lucidrains](https://github.com/lucidrains/).
+
+- [Metal](https://developer.apple.com/metal): an [implementation](https://github.com/philipturner/metal-flash-attention) in Metal by Philip Turner. This ports FlashAttention to mobile GPU architectures such as Apple silicon.