wolfecameron / nanoMoE
An extension of the nanoGPT repository for training small MOE models.
☆129Updated last month
Alternatives and similar repositories for nanoMoE:
Users that are interested in nanoMoE are comparing it to the libraries listed below
- ☆169Updated 2 months ago
- minimal GRPO implementation from scratch☆72Updated last month
- PyTorch building blocks for the OLMo ecosystem☆197Updated this week
- Memory layers use a trainable key-value lookup mechanism to add extra parameters to a model without increasing FLOPs. Conceptually, spars…☆317Updated 4 months ago
- A MAD laboratory to improve AI architecture designs 🧪☆111Updated 4 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆105Updated 5 months ago
- Understand and test language model architectures on synthetic tasks.☆192Updated last month
- code for training & evaluating Contextual Document Embedding models☆180Updated this week
- Code for "LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding", ACL 2024☆286Updated last week
- Normalized Transformer (nGPT)☆168Updated 5 months ago
- Manage scalable open LLM inference endpoints in Slurm clusters☆254Updated 9 months ago
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆240Updated this week
- Implementation of 🥥 Coconut, Chain of Continuous Thought, in Pytorch☆164Updated 3 months ago
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆230Updated 2 months ago
- ☆175Updated 4 months ago
- A project to improve skills of large language models☆283Updated this week
- Cold Compress is a hackable, lightweight, and open-source toolkit for creating and benchmarking cache compression methods built on top of…☆128Updated 8 months ago
- The official evaluation suite and dynamic data release for MixEval.☆235Updated 5 months ago
- A framework to study AI models in Reasoning, Alignment, and use of Memory (RAM).☆217Updated 2 weeks ago
- Single File, Single GPU, From Scratch, Efficient, Full Parameter Tuning library for "RL for LLMs"☆385Updated 2 weeks ago
- Exploring Applications of GRPO☆182Updated this week
- Code for studying the super weight in LLM☆98Updated 4 months ago
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆177Updated last week
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆59Updated 2 months ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆126Updated 4 months ago
- ☆71Updated this week
- Q-GaLore: Quantized GaLore with INT4 Projection and Layer-Adaptive Low-Rank Gradients.☆197Updated 9 months ago
- [ICML 2024] CLLMs: Consistency Large Language Models☆391Updated 5 months ago
- Official repository for "Scaling Retrieval-Based Langauge Models with a Trillion-Token Datastore".☆196Updated 2 weeks ago
- ☆185Updated this week