PyRank

Long Context Attention Python Packages

Python packages with the GitHub topic long-context-attention. Sorted by monthly downloads, with stars and forks.
lucidrains
block-recurrent-transformer-pytorch

Implementation of Block Recurrent Transformer - Pytorch

949 226 20
lucidrains
flash-attention-jax

Implementation of Flash Attention in Jax

257 230 25