CHARM-Tx / linear_mem_attention_pytorch

Unofficially Implements https://arxiv.org/abs/2112.05682 to get Linear Memory Cost on Attention for PyTorch
12Updated 3 years ago

Alternatives and similar repositories for linear_mem_attention_pytorch:

Users that are interested in linear_mem_attention_pytorch are comparing it to the libraries listed below