Alternatives and similar repositories for nanoMoE

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

• huggingface / picotron_tutorial
  ☆225Updated 3 weeks ago
• changjonathanc / flex-nano-vllm
  FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.
  ☆302Updated last week
• facebookresearch / memory
  Memory layers use a trainable key-value lookup mechanism to add extra parameters to a model without increasing FLOPs. Conceptually, spars…
  ☆355Updated 11 months ago
• facebookresearch / LayerSkip
  Code for "LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding", ACL 2024
  ☆344Updated 6 months ago
• HazyResearch / lolcats
  Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"
  ☆249Updated 9 months ago
• fangyuan-ksgk / Tiny-GRPO
  minimal GRPO implementation from scratch
  ☆99Updated 8 months ago
• MekkCyber / TritonAcademy
  A repository to unravel the language of GPUs, making their kernel conversations easy to understand
  ☆195Updated 5 months ago
• huggingface / kernels
  Load compute kernels from the Hub
  ☆326Updated this week
• microsoft / ArchScale
  Simple & Scalable Pretraining for Neural Architecture Research
  ☆298Updated 2 weeks ago
• shangshang-wang / Tina
  Tina: Tiny Reasoning Models via LoRA
  ☆304Updated last month
• allenai / OLMo-core
  PyTorch building blocks for the OLMo ecosystem
  ☆317Updated this week
• NVIDIA / ngpt
  Normalized Transformer (nGPT)
  ☆192Updated 11 months ago
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆271Updated last week
• NVlabs / Minitron
  A family of compressed models obtained via pruning and knowledge distillation
  ☆355Updated last week
• NVlabs / hymba
  ☆201Updated 11 months ago
• hkproj / triton-flash-attention
  ☆216Updated 10 months ago
• ServiceNow / PipelineRL
  A scalable asynchronous reinforcement learning implementation with in-flight weight updates.
  ☆291Updated this week
• rasbt / dora-from-scratch
  LoRA and DoRA from Scratch Implementations
  ☆211Updated last year
• NVIDIA / Star-Attention
  Efficient LLM Inference over Long Sequences
  ☆390Updated 4 months ago
• CASE-Lab-UMD / LLM-Drop
  The official implementation of the paper "What Matters in Transformers? Not All Attention is Needed".
  ☆179Updated 7 months ago
• ZHZisZZ / dllm
  dLLM: Simple Diffusion Language Modeling
  ☆529Updated this week
• joey00072 / nanoGRPO
  nanoGRPO is a lightweight implementation of Group Relative Policy Optimization (GRPO)
  ☆125Updated 6 months ago
• brendanhogan / DeepSeekRL-Extended
  Exploring Applications of GRPO
  ☆248Updated 2 months ago
• ZihanWang314 / CoE
  Chain of Experts (CoE) enables communication between experts within Mixture-of-Experts (MoE) models
  ☆223Updated last week
• NVIDIA-NeMo / Skills
  A project to improve skills of large language models
  ☆608Updated this week
• jxmorris12 / cde
  code for training & evaluating Contextual Document Embedding models
  ☆200Updated 6 months ago
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆171Updated 4 months ago
• astramind-ai / Mixture-of-depths
  Unofficial implementation for the paper "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"
  ☆175Updated last year
• snowflakedb / ArcticTraining
  ArcticTraining is a framework designed to simplify and accelerate the post-training process for large language models (LLMs)
  ☆241Updated this week
• tokenbender / avataRL
  rl from zero pretrain, can it be done? yes.
  ☆280Updated last month