wolfecameron / nanoMoELinks
An extension of the nanoGPT repository for training small MOE models.
☆185Updated 6 months ago
Alternatives and similar repositories for nanoMoE
Users that are interested in nanoMoE are comparing it to the libraries listed below
Sorting:
- ☆216Updated 7 months ago
- FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.☆269Updated last month
- PyTorch building blocks for the OLMo ecosystem☆286Updated this week
- Code for "LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding", ACL 2024☆334Updated 4 months ago
- Load compute kernels from the Hub☆271Updated this week
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆193Updated 3 months ago
- LoRA and DoRA from Scratch Implementations☆211Updated last year
- Normalized Transformer (nGPT)☆188Updated 9 months ago
- Memory layers use a trainable key-value lookup mechanism to add extra parameters to a model without increasing FLOPs. Conceptually, spars…☆343Updated 9 months ago
- minimal GRPO implementation from scratch☆96Updated 6 months ago
- ☆196Updated 9 months ago
- ☆199Updated 8 months ago
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆265Updated last month
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆245Updated 7 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆157Updated 2 months ago
- Efficient LLM Inference over Long Sequences☆391Updated 2 months ago
- Single File, Single GPU, From Scratch, Efficient, Full Parameter Tuning library for "RL for LLMs"☆524Updated 2 months ago
- code for training & evaluating Contextual Document Embedding models☆197Updated 4 months ago
- Tina: Tiny Reasoning Models via LoRA☆281Updated last month
- Simple & Scalable Pretraining for Neural Architecture Research☆291Updated 3 weeks ago
- Exploring Applications of GRPO☆247Updated 2 weeks ago
- A compact LLM pretrained in 9 days by using high quality data☆323Updated 5 months ago
- The official implementation of the paper "What Matters in Transformers? Not All Attention is Needed".☆175Updated 5 months ago
- A project to improve skills of large language models☆553Updated this week
- Open-source framework for the research and development of foundation models.☆419Updated this week
- rl from zero pretrain, can it be done? yes.☆265Updated 3 weeks ago
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)☆161Updated 5 months ago
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆181Updated 2 months ago
- ArcticTraining is a framework designed to simplify and accelerate the post-training process for large language models (LLMs)☆208Updated last week
- Q-GaLore: Quantized GaLore with INT4 Projection and Layer-Adaptive Low-Rank Gradients.☆200Updated last year