zms1999 / SmartMoE
A MoE impl for PyTorch, [ATC'23] SmartMoE
☆62Updated last year
Alternatives and similar repositories for SmartMoE:
Users that are interested in SmartMoE are comparing it to the libraries listed below
- ATC23 AE☆45Updated last year
- Train llm (bloom, llama, baichuan2-7b, chatglm3-6b) with deepspeed pipeline mode. Faster than zero/zero++/fsdp.☆95Updated last year
- Skywork-MoE: A Deep Dive into Training Techniques for Mixture-of-Experts Language Models☆132Updated 10 months ago
- ☆79Updated last year
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆80Updated 3 weeks ago
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆99Updated last year
- ☆131Updated 2 months ago
- Repository of LV-Eval Benchmark☆65Updated 8 months ago
- ☆74Updated 3 weeks ago
- Official implementation of ICML 2024 paper "ExCP: Extreme LLM Checkpoint Compression via Weight-Momentum Joint Shrinking".☆47Updated 9 months ago
- ☆109Updated 5 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆69Updated 10 months ago
- A flexible and efficient training framework for large-scale alignment tasks☆345Updated 2 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆97Updated last month
- code for Scaling Laws of RoPE-based Extrapolation☆73Updated last year
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)☆103Updated last month
- [ICML'24] The official implementation of “Rethinking Optimization and Architecture for Tiny Language Models”☆121Updated 3 months ago
- A simple calculation for LLM MFU.☆37Updated 2 months ago
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated last year
- ☆82Updated 3 years ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆209Updated 8 months ago
- A prototype repo for hybrid training of pipeline parallel and distributed data parallel with comments on core code snippets. Feel free to…☆55Updated last year
- ☆63Updated 5 months ago
- Trinity-RFT is a general-purpose, flexible and scalable framework designed for reinforcement fine-tuning (RFT) of large language models (…☆71Updated last week
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆39Updated last year
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆69Updated last year
- A more efficient GLM implementation!☆55Updated 2 years ago
- Mixture-of-Experts (MoE) Language Model☆186Updated 8 months ago
- This is a personal reimplementation of Google's Infini-transformer, utilizing a small 2b model. The project includes both model and train…☆56Updated last year
- Efficient Mixture of Experts for LLM Paper List☆64Updated 4 months ago