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

3K 225 20
lucidrains
flash-attention-jax

Flash Attention - in Jax

406 228 25