Rare find

Memory Efficient Attention. A way to run AI attention math without eating up all your computer's memory.

github.com/lucidrains/memory-efficient-attention-pytorch

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

July 2023
  • comments
  • readme
  • Release —0.1.6
  • further simplification
  • Release —0.1.5
  • add new trick from flash attention 2 that saves on division
  • remove 3.7 support
  • Release —0.1.4
  • bump version
  • allow for full attention mask in naive flash attention impl
March 2023
  • update
  • Release —0.1.2
  • 0.1.2
  • Merge pull request #7 from marksgraham/main
  • Import partial
December 2022
  • Release —0.1.1
  • fix tests
  • Release —0.1.0
  • 0.1.0
  • Merge pull request #6 from usryokousha/main
  • Added dropout support to memory efficient variant
November 2022
  • fix tests
  • torch 1.13 is broken for tests
  • Release —0.0.27
  • bring in the further simplification to flash attention that @tridao d…
July 2022
  • Release —0.0.26
  • fix cosine sim flash attention as well
  • Release —0.0.25
  • test out flash attention in GPT
  • Release —0.0.24