Search Packages

ring-attention-pytorch

Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch

megabyte-pytorch

Implementation of MEGABYTE, Predicting Million-byte Sequences with Multiscale Transformers, in Pytorch

recurrent-memory-transformer-pytorch

Implementation of Recurrent Memory Transformer, Neurips 2022 paper, in Pytorch

infini-transformer-pytorch

Implementation of Infini-Transformer in Pytorch

titans-memory

Titans in PyTorch: Infinite Context models that learn to remember. Faster than Transformers, smarter than RNNs, and fully HuggingFace compatible.

compactbench

Open benchmark for LLM context compaction methods — measures what survives when you replace conversation history with a compacted artifact. Multi-cycle drift, hidden ranked set.

perceiver-ar-pytorch

Implementation of Perceiver AR, Deepmind's new long-context attention network based on Perceiver architecture, in Pytorch

ringattention

RingAttention for Transformers with Arbitrarily Large Context.

nexusquant-kv

Training-free KV cache compression for LLMs. 10-33x compression via E8 lattice quantization + attention-aware token eviction. One line of code.

hard-needle

Semantically hard multi-needle long-context data generator. Stop testing LLMs with random-password needles.

kakeyalattice

Discrete Kakeya cover for LLM KV cache: D4/E8 nested-lattice quantisation realising a Kakeya-style tube-cover over the direction sphere. 2.4x-2.8x compression at <1% perplexity loss on Qwen3, Llama-3, DeepSeek, GLM-4, Gemma. Drop-in transformers.DynamicCache. pip install kakeyalattice.

liah

needle in a haystack for LLMs

mcp-long-context-reader

A tool to help agents read and query long documents.

replm

Recursive Language Models — process arbitrarily long prompts by offloading context into a REPL with symbolic recursion via sub-LLM calls. Based on Zhang, Kraska & Khattab (2025).

shadowkv

shadow kv cache