1 Repositories
Python linear_mem_attention_pytorch Libraries
Attention for PyTorch with Linear Memory Footprint
Attention for PyTorch with Linear Memory Footprint Unofficially implements https://arxiv.org/abs/2112.05682 to get Linear Memory Cost on Attention (+
11 Jan 9, 2022