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
- Transformer related optimization, including BERT, GPT☆61Updated last year
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆88Updated 10 months ago
- ☆71Updated 5 months ago
- ☆127Updated 3 weeks ago
- PyTorch bindings for CUTLASS grouped GEMM.☆84Updated 2 weeks ago
- A collection of memory efficient attention operators implemented in the Triton language.☆229Updated 7 months ago
- FlagScale is a large model toolkit based on open-sourced projects.☆207Updated this week
- ☆140Updated 8 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆64Updated 7 months ago
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆47Updated 5 months ago
- Models and examples built with OneFlow☆96Updated 3 months ago
- Transformer related optimization, including BERT, GPT☆39Updated last year
- A MoE impl for PyTorch, [ATC'23] SmartMoE☆60Updated last year
- ☆211Updated last year
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆400Updated 2 weeks ago
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated last year
- DashInfer is a native LLM inference engine aiming to deliver industry-leading performance atop various hardware architectures, including …☆208Updated this week
- Train llm (bloom, llama, baichuan2-7b, chatglm3-6b) with deepspeed pipeline mode. Faster than zero/zero++/fsdp.☆92Updated 11 months ago
- ☆78Updated 4 months ago
- ☆114Updated 10 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆89Updated 8 months ago
- Sequence-level 1F1B schedule for LLMs.☆17Updated 7 months ago
- Inferflow is an efficient and highly configurable inference engine for large language models (LLMs).☆236Updated 10 months ago
- A quantization algorithm for LLM☆108Updated 6 months ago
- Transformer related optimization, including BERT, GPT☆17Updated last year
- Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers☆204Updated 4 months ago
- Official implementation of ICML 2024 paper "ExCP: Extreme LLM Checkpoint Compression via Weight-Momentum Joint Shrinking".☆44Updated 6 months ago
- A flexible and efficient training framework for large-scale alignment tasks☆272Updated this week
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆33Updated 4 months ago
- ☆13Updated 9 months ago