Ascend / AscendSpeed
☆76Updated last year
Alternatives and similar repositories for AscendSpeed:
Users that are interested in AscendSpeed are comparing it to the libraries listed below
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆93Updated 11 months ago
- FlagScale is a large model toolkit based on open-sourced projects.☆223Updated this week
- ☆127Updated last month
- PyTorch bindings for CUTLASS grouped GEMM.☆94Updated last month
- A collection of memory efficient attention operators implemented in the Triton language.☆237Updated 8 months ago
- Transformer related optimization, including BERT, GPT☆59Updated last year
- ☆140Updated 9 months ago
- ☆67Updated 2 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆92Updated 9 months ago
- Models and examples built with OneFlow☆96Updated 4 months ago
- A MoE impl for PyTorch, [ATC'23] SmartMoE☆61Updated last year
- Odysseus: Playground of LLM Sequence Parallelism☆64Updated 8 months ago
- ☆101Updated 6 months ago
- Transformer related optimization, including BERT, GPT☆17Updated last year
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆35Updated 5 months ago
- Transformer related optimization, including BERT, GPT☆39Updated 2 years ago
- ☆81Updated 5 months ago
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆51Updated 6 months ago
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated last year
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆424Updated this week
- ☆14Updated 10 months ago
- ATC23 AE☆45Updated last year
- A flexible and efficient training framework for large-scale alignment tasks☆303Updated last week
- Train llm (bloom, llama, baichuan2-7b, chatglm3-6b) with deepspeed pipeline mode. Faster than zero/zero++/fsdp.☆92Updated last year
- ☆84Updated last year
- ☆40Updated this week
- A quantization algorithm for LLM☆132Updated 8 months ago
- ☆172Updated 4 months ago
- Triton Documentation in Chinese Simplified / Triton 中文文档☆54Updated last month