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
- ATC23 AE☆45Updated last year
- [ICLR 2025] PEARL: parallel speculative decoding with adaptive draft length☆54Updated last week
- ☆78Updated last year
- Skywork-MoE: A Deep Dive into Training Techniques for Mixture-of-Experts Language Models☆129Updated 9 months ago
- Train llm (bloom, llama, baichuan2-7b, chatglm3-6b) with deepspeed pipeline mode. Faster than zero/zero++/fsdp.☆94Updated last year
- code for Scaling Laws of RoPE-based Extrapolation☆70Updated last year
- Official implementation of ICML 2024 paper "ExCP: Extreme LLM Checkpoint Compression via Weight-Momentum Joint Shrinking".☆47Updated 8 months ago
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆93Updated last year
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)☆86Updated 5 months ago
- CLongEval: A Chinese Benchmark for Evaluating Long-Context Large Language Models☆40Updated last year
- ☆124Updated 2 weeks ago
- ☆60Updated 3 months ago
- [ICML'24] The official implementation of “Rethinking Optimization and Architecture for Tiny Language Models”☆121Updated 2 months ago
- ☆64Updated 3 months ago
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated last year
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆69Updated last year
- Repository of LV-Eval Benchmark☆59Updated 6 months ago
- The Official Implementation of Ada-KV: Optimizing KV Cache Eviction by Adaptive Budget Allocation for Efficient LLM Inference☆66Updated last month
- Efficient Mixture of Experts for LLM Paper List☆45Updated 3 months ago
- A flexible and efficient training framework for large-scale alignment tasks☆331Updated last month
- This is a personal reimplementation of Google's Infini-transformer, utilizing a small 2b model. The project includes both model and train…☆56Updated 11 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆66Updated 9 months ago
- [ACL 2024] Long-Context Language Modeling with Parallel Encodings☆153Updated 9 months ago
- ☆100Updated 11 months ago
- ☆80Updated 2 years ago
- Low-bit optimizers for PyTorch☆125Updated last year
- A LLaMA1/LLaMA12 Megatron implement.☆28Updated last year
- An easy-to-use package for implementing SmoothQuant for LLMs☆94Updated 10 months ago
- Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers☆207Updated 7 months ago
- ☆29Updated 6 months ago