ashwin-999 / memory-efficient-attention-pytorch Goto Github PK
View Code? Open in Web Editor NEWThis project forked from lucidrains/memory-efficient-attention-pytorch
Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"
License: MIT License