kwai / Megatron-Kwai
[USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Parallelism
☆51Updated 6 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.☆94Updated last month
- ☆83Updated 3 months ago
- ☆101Updated 6 months ago
- ☆72Updated 3 years ago
- nnScaler: Compiling DNN models for Parallel Training☆93Updated last week
- ☆81Updated 5 months ago
- A baseline repository of Auto-Parallelism in Training Neural Networks☆142Updated 2 years ago
- Sequence-level 1F1B schedule for LLMs.☆16Updated last month
- ☆79Updated 2 months ago
- ☆140Updated 9 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆237Updated 8 months ago
- A fast communication-overlapping library for tensor parallelism on GPUs.☆296Updated 3 months ago
- ATC23 AE☆45Updated last year
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆78Updated 3 months ago
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆93Updated 11 months ago
- ☆24Updated 6 months ago
- High performance Transformer implementation in C++.☆102Updated last month
- PyTorch distributed training acceleration framework☆39Updated last week
- Curated collection of papers in MoE model inference☆64Updated this week
- A tiny yet powerful LLM inference system tailored for researching purpose. vLLM-equivalent performance with only 2k lines of code (2% of …☆138Updated 7 months ago
- Dynamic Memory Management for Serving LLMs without PagedAttention☆290Updated this week
- ☆142Updated last month
- Zero Bubble Pipeline Parallelism☆336Updated last week
- flash attention tutorial written in python, triton, cuda, cutlass☆260Updated last month
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆424Updated this week
- Chimera: Efficiently Training Large-Scale Neural Networks with Bidirectional Pipelines.☆57Updated last year
- This repository is established to store personal notes and annotated papers during daily research.☆109Updated this week
- ☆110Updated 11 months ago
- ☆76Updated 2 years ago
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆53Updated 6 months ago