Ascend / AscendSpeed
☆70Updated 9 months ago
Related projects: ⓘ
- Transformer related optimization, including BERT, GPT☆58Updated 11 months ago
- FlagScale is a large model toolkit based on open-sourced projects.☆129Updated this week
- ☆123Updated 3 months ago
- Models and examples built with OneFlow☆94Updated last month
- ☆105Updated this week
- Transformer related optimization, including BERT, GPT☆39Updated last year
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆309Updated this week
- A collection of memory efficient attention operators implemented in the Triton language.☆205Updated 3 months ago
- ☆140Updated 4 months ago
- A MoE impl for PyTorch, [ATC'23] SmartMoE☆56Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆57Updated 2 months ago
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆71Updated 6 months ago
- Transformer related optimization, including BERT, GPT☆17Updated last year
- An easy-to-use package for implementing SmoothQuant for LLMs☆78Updated 4 months ago
- ☆60Updated last month
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆41Updated last month
- export llama to onnx☆91Updated 3 months ago
- Train llm (bloom, llama, baichuan2-7b, chatglm3-6b) with deepspeed pipeline mode. Faster than zero/zero++/fsdp.☆90Updated 7 months ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆20Updated last week
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated 9 months ago
- ☆205Updated last year
- A more efficient GLM implementation!☆54Updated last year
- A quantization algorithm for LLM☆98Updated 2 months ago
- ☆82Updated last year
- Simple Dynamic Batching Inference☆145Updated 2 years ago
- ATC23 AE☆42Updated last year
- Ascend PyTorch adapter (torch_npu). Mirror of https://gitee.com/ascend/pytorch☆226Updated this week
- ☆108Updated 6 months ago
- ☆251Updated last week
- ☆145Updated last month