Alternatives and similar repositories for UMoE

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

• OpenNLPLab / HGRN2
  HGRN2: Gated Linear RNNs with State Expansion
  ☆56Updated last year
• tianyi-lab / R2-T2
  [ICML 2025] Code for "R2-T2: Re-Routing in Test-Time for Multimodal Mixture-of-Experts"
  ☆17Updated 10 months ago
• MambaMixer / M2
  ☆48Updated last year
• fangyuan-ksgk / selective-attention-transformer
  Unofficial Implementation of Selective Attention Transformer
  ☆20Updated last year
• smonsays / hypernetwork-attention
  Official code for the paper "Attention as a Hypernetwork"
  ☆47Updated last year
• g-luo / vlm_cross_modal_reps
  Official PyTorch Implementation for Vision-Language Models Create Cross-Modal Task Representations, ICML 2025
  ☆31Updated 9 months ago
• caojiaolong / Awesome-Mamba
  Collect papers about Mamba (a selective state space model).
  ☆14Updated last year
• PKU-ML / non_neg
  Official Code for ICLR 2024 Paper: Non-negative Contrastive Learning
  ☆46Updated last year
• qiuzh20 / RMoE
  Official implementation of RMoE (Layerwise Recurrent Router for Mixture-of-Experts)
  ☆29Updated last year
• ChuanyangZheng / L2ViT
  Official PyTorch implementation of The Linear Attention Resurrection in Vision Transformer
  ☆15Updated last year
• locuslab / llava-token-compression
  ☆46Updated last year
• Doraemonzzz / hgru2-pytorch
  ☆24Updated 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 11 months ago
• alhojel / visual_task_vectors
  ☆40Updated last year
• krafton-ai / mambaformer-icl
  MambaFormer in-context learning experiments and implementation for https://arxiv.org/abs/2402.04248
  ☆58Updated last year
• UCDvision / NOLA
  Code for NOLA, an implementation of "nola: Compressing LoRA using Linear Combination of Random Basis"
  ☆57Updated last year
• roymiles / VeLoRA
  [NeurIPS 2024] VeLoRA : Memory Efficient Training using Rank-1 Sub-Token Projections
  ☆21Updated last year
• HelmholtzAI-FZJ / flex_gen
  ☆19Updated last year
• 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…
  ☆29Updated 6 months ago
• tim-lawson / skip-middle
  Learning to Skip the Middle Layers of Transformers
  ☆17Updated 5 months ago
• Benjamin-Walker / selective-ssms-and-linear-cdes
  Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)
  ☆15Updated last year
• oripress / EntropyEnigma
  Official code for the ICML 2024 paper "The Entropy Enigma: Success and Failure of Entropy Minimization"
  ☆55Updated last year
• epfml / pam
  ☆16Updated 2 years ago
• assafbk / DeciMamba
  DeciMamba: Exploring the Length Extrapolation Potential of Mamba (ICLR 2025)
  ☆32Updated 9 months ago
• ethanbar11 / ssm_2d
  More dimensions = More fun
  ☆26Updated last year
• BryceZhuo / PolyCom
  The official implementation of ICLR 2025 paper "Polynomial Composition Activations: Unleashing the Dynamics of Large Language Models".
  ☆17Updated 9 months ago
• ethanlshen / HierNet
  Code for "Are “Hierarchical” Visual Representations Hierarchical?" in NeurIPS Workshop for Symmetry and Geometry in Neural Representation…
  ☆22Updated 2 years ago
• NVlabs / ConvSSM
  ☆69Updated last year
• BatsResearch / ex2
  If CLIP Could Talk: Understanding Vision-Language Model Representations Through Their Preferred Concept Descriptions
  ☆17Updated last year