lucidrains / PEER-pytorch
Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind
β122Updated 7 months ago
Alternatives and similar repositories for PEER-pytorch:
Users that are interested in PEER-pytorch are comparing it to the libraries listed below
- Mixture of A Million Expertsβ42Updated 8 months ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"β224Updated last month
- Implementation of π₯₯ Coconut, Chain of Continuous Thought, in Pytorchβ161Updated 3 months ago
- Some preliminary explorations of Mamba's context scaling.β212Updated last year
- Griffin MQA + Hawk Linear RNN Hybridβ85Updated 11 months ago
- Understand and test language model architectures on synthetic tasks.β185Updated 3 weeks ago
- β74Updated 7 months ago
- Token Omission Via Attentionβ124Updated 5 months ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clustersβ125Updated 4 months ago
- Normalized Transformer (nGPT)β164Updated 4 months ago
- Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"β97Updated 6 months ago
- Implementation of Infini-Transformer in Pytorchβ110Updated 2 months ago
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"β226Updated 2 months ago
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Ruleβ143Updated 2 weeks ago
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)β150Updated 3 months ago
- β182Updated this week
- supporting pytorch FSDP for optimizersβ80Updated 3 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.β103Updated 4 months ago
- β87Updated 6 months ago
- β65Updated last month
- This is the official repository for the paper "Flora: Low-Rank Adapters Are Secretly Gradient Compressors" in ICML 2024.β102Updated 9 months ago
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Modelsβ209Updated last month
- Explorations into the recently proposed Taylor Series Linear Attentionβ96Updated 7 months ago
- Official implementation of Phi-Mamba. A MOHAWK-distilled model (Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Modeβ¦β102Updated 6 months ago
- β81Updated last year
- [ICLR 2025] Official PyTorch implementation of "Forgetting Transformer: Softmax Attention with a Forget Gate"β80Updated last week
- β76Updated 8 months ago
- research impl of Native Sparse Attention (2502.11089)β53Updated last month
- PyTorch implementation of models from the Zamba2 series.β178Updated 2 months ago
- This repo is based on https://github.com/jiaweizzhao/GaLoreβ26Updated 6 months ago