kwai / Megatron-Kwai
[USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Parallelism
☆51Updated 7 months ago
Alternatives and similar repositories for Megatron-Kwai:
Users that are interested in Megatron-Kwai are comparing it to the libraries listed below
- PyTorch bindings for CUTLASS grouped GEMM.☆108Updated 2 months ago
- ☆90Updated 4 months ago
- ☆125Updated 2 weeks ago
- nnScaler: Compiling DNN models for Parallel Training☆103Updated last month
- ☆72Updated 3 years ago
- ☆88Updated 6 months ago
- High performance Transformer implementation in C++.☆109Updated 2 months ago
- PyTorch distributed training acceleration framework☆44Updated last month
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆93Updated last year
- A tiny yet powerful LLM inference system tailored for researching purpose. vLLM-equivalent performance with only 2k lines of code (2% of …☆151Updated 8 months ago
- ☆81Updated 2 years ago
- A baseline repository of Auto-Parallelism in Training Neural Networks☆143Updated 2 years ago
- Sequence-level 1F1B schedule for LLMs.☆16Updated 3 months ago
- ATC23 AE☆45Updated last year
- ☆139Updated 11 months ago
- ☆31Updated 7 months ago
- Curated collection of papers in MoE model inference☆110Updated last month
- Since the emergence of chatGPT in 2022, the acceleration of Large Language Model has become increasingly important. Here is a list of pap…☆235Updated 2 weeks ago
- A collection of memory efficient attention operators implemented in the Triton language.☆253Updated 9 months ago
- ☆79Updated 3 months ago
- ☆52Updated 11 months ago
- Zero Bubble Pipeline Parallelism☆373Updated 2 weeks ago
- ☆64Updated 3 months ago
- ☆127Updated 3 months ago
- Chimera: bidirectional pipeline parallelism for efficiently training large-scale models.☆60Updated this week
- This repository is established to store personal notes and annotated papers during daily research.☆115Updated this week
- A ChatGPT(GPT-3.5) & GPT-4 Workload Trace to Optimize LLM Serving Systems☆153Updated 5 months ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆35Updated 3 weeks ago
- ☆145Updated 2 months ago