YeonwooSung / Pytorch_mixture-of-experts
PyTorch implementation of moe, which stands for mixture of experts
☆32Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for Pytorch_mixture-of-experts
- PyTorch implementation of Soft MoE by Google Brain in "From Sparse to Soft Mixtures of Experts" (https://arxiv.org/pdf/2308.00951.pdf)☆64Updated last year
- Implementation of TableFormer, Robust Transformer Modeling for Table-Text Encoding, in Pytorch☆35Updated 2 years ago
- Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"☆91Updated last month
- ☆36Updated last year
- A Closer Look into Mixture-of-Experts in Large Language Models☆38Updated 3 months ago
- Implementation of MoE Mamba from the paper: "MoE-Mamba: Efficient Selective State Space Models with Mixture of Experts" in Pytorch and Ze…☆83Updated this week
- The official implementation of the paper "What Matters in Transformers? Not All Attention is Needed".☆120Updated 2 weeks ago
- Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts☆108Updated 3 weeks ago
- 32 times longer context window than vanilla Transformers and up to 4 times longer than memory efficient Transformers.☆42Updated last year
- Implementation of Infini-Transformer in Pytorch☆104Updated last month
- PyTorch implementation of "From Sparse to Soft Mixtures of Experts"☆44Updated last year
- [NeurIPS-2024] 📈 Scaling Laws with Vocabulary: Larger Models Deserve Larger Vocabularies https://arxiv.org/abs/2407.13623☆67Updated last month
- ☆47Updated 3 months ago
- Large Scale Distributed Model Training strategy with Colossal AI and Lightning AI☆58Updated last year
- Contextual Position Encoding but with some custom CUDA Kernels https://arxiv.org/abs/2405.18719☆19Updated 5 months ago
- InstructRAG: Instructing Retrieval-Augmented Generation via Self-Synthesized Rationales☆51Updated 3 weeks ago
- Linear Attention Sequence Parallelism (LASP)☆64Updated 5 months ago
- ☆34Updated 8 months ago
- ☆97Updated 8 months ago
- Implementation of the general framework for AMIE, from the paper "Towards Conversational Diagnostic AI", out of Google Deepmind☆52Updated last month
- [ICLR 2023] "Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers" by Tianlong Chen*, Zhenyu Zhang*, Ajay Jaiswal…☆44Updated last year
- Implementation of CALM from the paper "LLM Augmented LLMs: Expanding Capabilities through Composition", out of Google Deepmind☆168Updated last month
- Pytorch Implementation of the sparse attention from the paper: "Generating Long Sequences with Sparse Transformers"☆59Updated this week
- A huge dataset for Document Visual Question Answering☆13Updated 3 months ago
- Implementation of a Transformer using ReLA (Rectified Linear Attention) from https://arxiv.org/abs/2104.07012☆49Updated 2 years ago
- Implementation of a Light Recurrent Unit in Pytorch☆46Updated last month
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆79Updated last year
- several types of attention modules written in PyTorch☆39Updated last month
- Randomized Positional Encodings Boost Length Generalization of Transformers☆79Updated 7 months ago