Efficient Infinite Context Transformers with Infini-attention Pytorch Implementation + QwenMoE Implementation + Training Script + 1M context keypass retrieval
☆100May 9, 2024Updated 2 years ago
Alternatives and similar repositories for Infini-Attention
Users that are interested in Infini-Attention are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Unofficial PyTorch/🤗Transformers(Gemma/Llama3) implementation of Leave No Context Behind: Efficient Infinite Context Transformers with I…☆375Apr 23, 2024Updated 2 years ago
- PyTorch implementation of Infini-Transformer from "Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention…☆301May 4, 2024Updated 2 years ago
- Pytorch implementation of https://arxiv.org/html/2404.07143v1☆21Apr 13, 2024Updated 2 years ago
- An unofficial pytorch implementation of 'Efficient Infinite Context Transformers with Infini-attention'☆56Aug 19, 2024Updated 2 years ago
- Implementation of the paper: "Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention" from Google in pyTO…☆59Oct 26, 2026Updated last week
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- ☆16Jul 29, 2025Updated last year
- [ICML 24 NGSM workshop] Associative Recurrent Memory Transformer implementation and scripts for training and evaluation☆68Aug 14, 2026Updated 2 months ago
- Quasar Attention Continuous-Time Memory Dynamics for Scaling Models Context to Millions of Tokens☆30Mar 26, 2026Updated 7 months ago
- This is a personal reimplementation of Google's Infini-transformer, utilizing a small 2b model. The project includes both model and train…☆58Apr 20, 2024Updated 2 years ago
- GPT* - Training faster small transformers using ALiBi, Parallel Residual Connections and more!☆20Oct 29, 2022Updated 4 years ago
- Muon in Int8 Precision Made Possible☆20Jun 18, 2026Updated 4 months ago
- Landing repository for the paper "Softpick: No Attention Sink, No Massive Activations with Rectified Softmax"☆92Sep 12, 2025Updated last year
- To mitigate position bias in LLMs, especially in long-context scenarios, we scale only one dimension of LLMs, reducing position bias and …☆13Jun 18, 2024Updated 2 years ago
- Set of scripts to finetune LLMs☆39Mar 30, 2024Updated 2 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- DPO, but faster 🚀☆51Dec 6, 2024Updated last year
- BEAVER: A Training-Free Hierarchical Prompt Compression Method via Structure-Aware Page Selection☆18Mar 30, 2026Updated 7 months ago
- [ICML'24] Data and code for our paper "Training-Free Long-Context Scaling of Large Language Models"☆451Oct 16, 2024Updated 2 years ago
- ☆52Oct 29, 2024Updated 2 years ago
- ☆52Oct 9, 2026Updated 3 weeks ago
- ☆16Aug 26, 2026Updated 2 months ago
- Open source repo for simple SOTA method.☆14Apr 21, 2026Updated 6 months ago
- some common Huggingface transformers in maximal update parametrization (µP)☆88Mar 14, 2022Updated 4 years ago
- Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models☆345Feb 23, 2025Updated last year
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Official PyTorch implementation of "Latent Reasoning in TRMs is Secretly a Policy Improvement Operator" (ICML 2026)☆26May 29, 2026Updated 5 months ago
- an implementation of paper"Retentive Network: A Successor to Transformer for Large Language Models" https://arxiv.org/pdf/2307.08621.pdf☆11Jul 25, 2023Updated 3 years ago
- Implementation of Strassen attention, from Kozachinskiy et al. of National Center of AI in Chile☆41Jul 8, 2025Updated last year
- Depth-Bounded PCFG Induction☆13Apr 19, 2019Updated 7 years ago
- Code for SFT and RL☆12Jun 22, 2025Updated last year
- Word embeddings from PPMI-weighted and dirichlet-smoothed co-occurrence matrices☆10Aug 3, 2020Updated 6 years ago
- Research work aimed at addressing the problem of modeling infinite-length context☆53Dec 18, 2025Updated 10 months ago
- Implementation of Recurrent Memory Transformer, Neurips 2022 paper, in Pytorch☆425Jan 6, 2025Updated last year
- ETR: An Efficient Transformer for Re-ranking in Visual Place Recognition (WACV 2023)☆17Nov 10, 2022Updated 3 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- ☆53May 13, 2024Updated 2 years ago
- Minimal and highly hackable implementation of Looped Transformers with GPT☆27Mar 8, 2026Updated 7 months ago
- High performance RMSNorm Implement by using SM Core Storage(Registers and Shared Memory)☆31Jan 22, 2026Updated 9 months ago
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning☆155Feb 25, 2026Updated 8 months ago
- Official implementation of ECCV24 paper: POA☆24Aug 8, 2024Updated 2 years ago
- ☆19Jun 3, 2026Updated 5 months ago
- Implementation of the LongRoPE: Extending LLM Context Window Beyond 2 Million Tokens Paper☆154Jul 20, 2024Updated 2 years ago