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: 9
Downloads last week: 563
Downloads last month: 734