zms1999 / SmartMoE
A MoE impl for PyTorch, [ATC'23] SmartMoE
☆61Updated last year
Alternatives and similar repositories for SmartMoE:
Users that are interested in SmartMoE are comparing it to the libraries listed below
- ☆76Updated last year
- ATC23 AE☆45Updated last year
- Train llm (bloom, llama, baichuan2-7b, chatglm3-6b) with deepspeed pipeline mode. Faster than zero/zero++/fsdp.☆92Updated last year
- ☆97Updated 6 months ago
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)☆84Updated 4 months ago
- Official implementation of ICML 2024 paper "ExCP: Extreme LLM Checkpoint Compression via Weight-Momentum Joint Shrinking".☆46Updated 7 months ago
- Skywork-MoE: A Deep Dive into Training Techniques for Mixture-of-Experts Language Models☆128Updated 8 months ago
- code for Scaling Laws of RoPE-based Extrapolation☆70Updated last year
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆93Updated 11 months ago
- Repository of LV-Eval Benchmark☆58Updated 5 months ago
- [ICML'24] The official implementation of “Rethinking Optimization and Architecture for Tiny Language Models”☆120Updated last month
- [ICLR 2025] PEARL: parallel speculative decoding with adaptive draft length☆37Updated 5 months ago
- This is a personal reimplementation of Google's Infini-transformer, utilizing a small 2b model. The project includes both model and train…☆55Updated 9 months ago
- A flexible and efficient training framework for large-scale alignment tasks☆297Updated this week
- Code associated with the paper **Draft & Verify: Lossless Large Language Model Acceleration via Self-Speculative Decoding**☆157Updated this week
- Odysseus: Playground of LLM Sequence Parallelism☆64Updated 7 months ago
- ☆84Updated last year
- ☆28Updated 5 months ago
- CLongEval: A Chinese Benchmark for Evaluating Long-Context Large Language Models☆38Updated 11 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆93Updated last month
- [ACL 2024] Long-Context Language Modeling with Parallel Encodings☆153Updated 8 months ago
- Inferflow is an efficient and highly configurable inference engine for large language models (LLMs).☆237Updated 11 months ago
- ☆60Updated this week
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆69Updated last year
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated last year
- ☆99Updated 3 months ago
- ☆94Updated 10 months ago
- Mixture-of-Experts (MoE) Language Model☆184Updated 5 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆89Updated 8 months ago
- A LLaMA1/LLaMA12 Megatron implement.☆28Updated last year