PyRank

Long Context Attention Python Packages

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

Implementation of Block Recurrent Transformer - Pytorch

4K 226 20
lucidrains
flash-attention-jax

Implementation of Flash Attention in Jax

497 228 25