GitHub repository: AminRezaei0x443/memory-efficient-attention
Primary language: Python
Created: 2021-12-18T19:27:26Z
Description: Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch