Ascend / DeepSpeedLinks
☆15Updated last year
Alternatives and similar repositories for DeepSpeed
Users that are interested in DeepSpeed are comparing it to the libraries listed below
Sorting:
- LiBai(李白): A Toolbox for Large-Scale Distributed Parallel Training☆407Updated 2 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆156Updated 2 weeks ago
- An industrial extension library of pytorch to accelerate large scale model training☆49Updated 2 months ago
- ☆79Updated last year
- A collection of memory efficient attention operators implemented in the Triton language.☆282Updated last year
- ☆64Updated this week
- ☆129Updated 10 months ago
- FlagScale is a large model toolkit based on open-sourced projects.☆364Updated this week
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆479Updated last year
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆242Updated 2 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆582Updated 2 weeks ago
- ☆219Updated 2 years ago
- Ascend PyTorch adapter (torch_npu). Mirror of https://gitee.com/ascend/pytorch☆444Updated last month
- Pipeline Parallelism Emulation and Visualization☆69Updated 4 months ago
- ☆139Updated last year
- ☆148Updated 7 months ago
- Ascend TileLang adapter☆130Updated last week
- Zero Bubble Pipeline Parallelism☆433Updated 5 months ago
- Transformer related optimization, including BERT, GPT☆59Updated 2 years ago
- Tutorials for writing high-performance GPU operators in AI frameworks.☆132Updated 2 years ago
- InternEvo is an open-sourced lightweight training framework aims to support model pre-training without the need for extensive dependencie…☆411Updated 2 months ago
- ☆150Updated 9 months ago
- 青稞Talk☆156Updated this week
- 🤖FFPA: Extend FlashAttention-2 with Split-D, ~O(1) SRAM complexity for large headdim, 1.8x~3x↑🎉 vs SDPA EA.☆226Updated 2 months ago
- Megatron's multi-modal data loader☆252Updated last week
- ☆97Updated 7 months ago
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆116Updated last year
- Implementation of FlashAttention in PyTorch☆172Updated 9 months ago
- DeepLearning Framework Performance Profiling Toolkit☆292Updated 3 years ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆41Updated 8 months ago