memory-efficient-attention-pytorch

PyPI page
Home page
Author: Phil Wang
License: MIT
Summary: Memory Efficient Attention - Pytorch
Latest version: 0.1.6
Required dependencies: einops | torch

Downloads last day: 23
Downloads last week: 153
Downloads last month: 827