Alternatives and similar repositories for UMoE

Users that are interested in UMoE are comparing it to the libraries listed below

• tianyi-lab / R2-T2
  [ICML 2025] Code for "R2-T2: Re-Routing in Test-Time for Multimodal Mixture-of-Experts"
  ☆17Updated 9 months ago
• fangyuan-ksgk / selective-attention-transformer
  Unofficial Implementation of Selective Attention Transformer
  ☆20Updated last year
• g-luo / vlm_cross_modal_reps
  Official PyTorch Implementation for Vision-Language Models Create Cross-Modal Task Representations, ICML 2025
  ☆31Updated 8 months ago
• OpenNLPLab / HGRN2
  HGRN2: Gated Linear RNNs with State Expansion
  ☆56Updated last year
• HelmholtzAI-FZJ / flex_gen
  ☆19Updated 11 months ago
• tim-lawson / skip-middle
  Learning to Skip the Middle Layers of Transformers
  ☆16Updated 5 months ago
• MambaMixer / M2
  ☆48Updated last year
• locuslab / llava-token-compression
  ☆46Updated last year
• PKU-ML / non_neg
  Official Code for ICLR 2024 Paper: Non-negative Contrastive Learning
  ☆46Updated last year
• oripress / EntropyEnigma
  Official code for the ICML 2024 paper "The Entropy Enigma: Success and Failure of Entropy Minimization"
  ☆55Updated last year
• Eliyas0007 / Pytorch-Intention
  Unofficial implementation of paper : Exploring the Space of Key-Value-Query Models with Intention
  ☆12Updated 2 years ago
• VijayLingam95 / SVFT
  ☆34Updated 10 months ago
• ablghtianyi / ICL_Modular_Arithmetic
  ☆19Updated 9 months ago
• smonsays / hypernetwork-attention
  Official code for the paper "Attention as a Hypernetwork"
  ☆46Updated last year
• ExplainableML / fomo_in_flux
  Code and benchmark for the paper: "A Practitioner's Guide to Continual Multimodal Pretraining" [NeurIPS'24]
  ☆61Updated last year
• RobertCsordas / switchhead
  ☆17Updated 6 months ago
• pixeli99 / MixLN
  [ICLR 2025] Official Pytorch Implementation of "Mix-LN: Unleashing the Power of Deeper Layers by Combining Pre-LN and Post-LN" by Pengxia…
  ☆27Updated 5 months ago
• james-oldfield / muMoE
  [NeurIPS'24] Multilinear Mixture of Experts: Scalable Expert Specialization through Factorization
  ☆38Updated last year
• caojiaolong / Awesome-Mamba
  Collect papers about Mamba (a selective state space model).
  ☆14Updated last year
• assafbk / DeciMamba
  DeciMamba: Exploring the Length Extrapolation Potential of Mamba (ICLR 2025)
  ☆32Updated 8 months ago
• roymiles / VeLoRA
  [NeurIPS 2024] VeLoRA : Memory Efficient Training using Rank-1 Sub-Token Projections
  ☆21Updated last year
• formll / resolving-scaling-law-discrepancies
  ☆20Updated 2 months ago
• alhojel / visual_task_vectors
  ☆40Updated last year
• nasosger / MuToR
  [NeurIPS '25] Multi-Token Prediction Needs Registers
  ☆26Updated 3 weeks ago
• ethanlshen / HierNet
  Code for "Are “Hierarchical” Visual Representations Hierarchical?" in NeurIPS Workshop for Symmetry and Geometry in Neural Representation…
  ☆21Updated 2 years ago
• arubique / OCCAM
  This is an implementation of the paper "Are We Done with Object-Centric Learning?"
  ☆12Updated 3 months ago
• Benjamin-Walker / selective-ssms-and-linear-cdes
  Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)
  ☆15Updated last year
• Weixin-Liang / Mixture-of-Mamba
  ☆50Updated 11 months ago
• minhoooo1 / CatMAE
  CatMAE
  ☆14Updated 2 years ago
• s-sahoo / MuLAN
  [NeurIPS 2024, spotlight] Multivariate Learned Adaptive Noise for Diffusion Models
  ☆30Updated last year