A Pytorch implementation of Sparsely-Gated Mixture of Experts, for massively increasing the parameter count of language models
☆877Sep 13, 2023Updated 3 years ago
Alternatives and similar repositories for mixture-of-experts
Users that are interested in mixture-of-experts are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- PyTorch Re-Implementation of "The Sparsely-Gated Mixture-of-Experts Layer" by Noam Shazeer et al. https://arxiv.org/abs/1701.06538☆1,256Apr 19, 2024Updated 2 years ago
- Implementation of ST-Moe, the latest incarnation of MoE after years of research at Brain, in Pytorch☆390Jun 17, 2024Updated 2 years ago
- Implementation of Soft MoE, proposed by Brain's Vision team, in Pytorch☆349Apr 2, 2025Updated last year
- A fast MoE impl for PyTorch☆1,859Feb 10, 2025Updated last year
- A collection of AWESOME things about mixture-of-experts☆1,286Dec 8, 2024Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- ☆728Updated this week
- Tutel MoE: Optimized Mixture-of-Experts Library, Support GLM/Kimi/Gpt/DeepSeek/Qwen using FP8/NVFP4/MXFP4☆1,030Oct 21, 2026Updated last week
- Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts☆121Oct 17, 2024Updated 2 years ago
- A curated reading list of research in Mixture-of-Experts(MoE).☆672Oct 30, 2024Updated 2 years ago
- Implementation of Switch Transformers from the paper: "Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficien…☆148Oct 26, 2026Updated last week
- A family of open-sourced Mixture-of-Experts (MoE) Large Language Models☆1,702Mar 8, 2024Updated 2 years ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆845May 5, 2024Updated 2 years ago
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆827Jun 20, 2026Updated 4 months ago
- An implementation of local windowed attention for language modeling☆505Jul 16, 2025Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Code for "DAMEX: Dataset-aware Mixture-of-Experts for visual understanding of mixture-of-datasets", accepted at Neurips 2023 (Main confer…☆28Mar 29, 2024Updated 2 years ago
- ☆1,600Mar 25, 2026Updated 7 months ago
- ⛷️ LLaMA-MoE: Building Mixture-of-Experts from LLaMA with Continual Pre-training (EMNLP 2024)☆1,004Dec 6, 2024Updated last year
- Towards Understanding the Mixture-of-Experts Layer in Deep Learning☆37Dec 12, 2023Updated 2 years ago
- Implementation of Memformer, a Memory-augmented Transformer, in Pytorch☆125Nov 13, 2020Updated 5 years ago
- My own attempt at a long context genomics model, leveraging recent advances in long context attention modeling (Flash Attention + other h…☆54Jul 2, 2023Updated 3 years ago
- Implementation of 🌻 Mirasol, SOTA Multimodal Autoregressive model out of Google Deepmind, in Pytorch☆92Dec 22, 2023Updated 2 years ago
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,207Jul 9, 2026Updated 3 months ago
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆5,962Oct 23, 2026Updated last week
- 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.
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆189Jun 24, 2022Updated 4 years ago
- PyTorch implementation of Soft MoE by Google Brain in "From Sparse to Soft Mixtures of Experts" (https://arxiv.org/pdf/2308.00951.pdf)☆83Oct 5, 2023Updated 3 years ago
- PyTorch extensions for high performance and large scale training.☆3,404Apr 26, 2025Updated last year
- Ongoing research training transformer models at scale☆18,310Updated this week
- This package implements THOR: Transformer with Stochastic Experts.☆64Oct 7, 2021Updated 5 years ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways☆823Nov 9, 2022Updated 3 years ago
- Mamba SSM architecture☆18,966Jul 22, 2026Updated 3 months ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆56Jan 5, 2023Updated 3 years ago
- Triton-based implementation of Sparse Mixture of Experts.☆281Oct 3, 2025Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Implementation of the "the first large-scale multimodal mixture of experts models." from the paper: "Multimodal Contrastive Learning with…☆36Oct 10, 2026Updated 3 weeks ago
- Fast and memory-efficient exact attention☆25,235Updated this week
- Implementation of Fast Transformer in Pytorch☆176Aug 26, 2021Updated 5 years ago
- [ICCV23] Robust Mixture-of-Expert Training for Convolutional Neural Networks by Yihua Zhang, Ruisi Cai, Tianlong Chen, Guanhua Zhang, Hua…☆69Aug 27, 2023Updated 3 years ago
- Vector (and Scalar) Quantization, in Pytorch☆4,023Oct 24, 2026Updated last week
- Self contained pytorch implementation of a sinkhorn based router, for mixture of experts or otherwise☆40Aug 29, 2024Updated 2 years ago
- ☆279Oct 31, 2023Updated 3 years ago