Alternatives and similar repositories for mixture-of-experts:

Users that are interested in mixture-of-experts are comparing it to the libraries listed below

• davidmrau / mixture-of-experts
  PyTorch Re-Implementation of "The Sparsely-Gated Mixture-of-Experts Layer" by Noam Shazeer et al. https://arxiv.org/abs/1701.06538
  ☆1,027Updated 9 months ago
• lucidrains / st-moe-pytorch
  Implementation of ST-Moe, the latest incarnation of MoE after years of research at Brain, in Pytorch
  ☆299Updated 7 months ago
• microsoft / Tutel
  Tutel MoE: An Optimized Mixture-of-Experts Implementation
  ☆750Updated last week
• codecaution / Awesome-Mixture-of-Experts-Papers
  A curated reading list of research in Mixture-of-Experts(MoE).
  ☆558Updated 3 months ago
• XueFuzhao / awesome-mixture-of-experts
  A collection of AWESOME things about mixture-of-experts
  ☆1,034Updated last month
• google-research / vmoe
  ☆600Updated last week
• laekov / fastmoe
  A fast MoE impl for PyTorch
  ☆1,606Updated 6 months ago
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆731Updated 8 months ago
• lucidrains / rotary-embedding-torch
  Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch
  ☆619Updated 2 months ago
• lucidrains / soft-moe-pytorch
  Implementation of Soft MoE, proposed by Brain's Vision team, in Pytorch
  ☆257Updated 9 months ago
• pytorch-labs / attention-gym
  Helpful tools and examples for working with flex-attention
  ☆603Updated this week
• google-research / long-range-arena
  Long Range Arena for Benchmarking Efficient Transformers
  ☆740Updated last year
• ZhuiyiTechnology / roformer
  Rotary Transformer
  ☆867Updated 2 years ago
• radarFudan / Awesome-state-space-models
  Collection of papers on state-space models
  ☆572Updated this week
• haoliuhl / ringattention
  Large Context Attention
  ☆677Updated this week
• fla-org / flash-linear-attention
  🚀 Efficient implementations of state-of-the-art linear attention models in Pytorch and Triton
  ☆1,822Updated this week
• ofirpress / attention_with_linear_biases
  Code for the ALiBi method for transformer language models (ICLR 2022)
  ☆512Updated last year
• Haiyang-W / TokenFormer
  [ICLR2025] Official Implementation of TokenFormer: Rethinking Transformer Scaling with Tokenized Model Parameters
  ☆482Updated this week
• facebookresearch / multimodal
  TorchMultimodal is a PyTorch library for training state-of-the-art multimodal multi-task models at scale.
  ☆1,527Updated this week
• MrYxJ / calculate-flops.pytorch
  The calflops is designed to calculate FLOPs、MACs and Parameters in all various neural networks, such as Linear、 CNN、 RNN、 GCN、Transformer…
  ☆667Updated 7 months ago
• lucidrains / recurrent-memory-transformer-pytorch
  Implementation of Recurrent Memory Transformer, Neurips 2022 paper, in Pytorch
  ☆403Updated 3 weeks ago
• lucidrains / ring-attention-pytorch
  Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch
  ☆498Updated 3 months ago
• zhuzilin / ring-flash-attention
  Ring attention implementation with flash attention
  ☆660Updated last month
• lucidrains / FLASH-pytorch
  Implementation of the Transformer variant proposed in "Transformer Quality in Linear Time"
  ☆357Updated last year
• lucidrains / local-attention
  An implementation of local windowed attention for language modeling
  ☆408Updated last week
• jxhe / unify-parameter-efficient-tuning
  Implementation of paper "Towards a Unified View of Parameter-Efficient Transfer Learning" (ICLR 2022)
  ☆521Updated 2 years ago
• idiap / fast-transformers
  Pytorch library for fast transformer implementations
  ☆1,670Updated last year
• OpenNLPLab / TransnormerLLM
  Official implementation of TransNormerLLM: A Faster and Better LLM
  ☆238Updated last year
• AvivBick / awesome-ssm-ml
  Reading list for research topics in state-space models
  ☆257Updated last week
• bobby-he / simplified_transformers
  ☆287Updated last month