Ascend / AscendSpeed
☆74Updated 11 months ago
Related projects ⓘ
Alternatives and complementary repositories for AscendSpeed
- PyTorch bindings for CUTLASS grouped GEMM.☆68Updated 4 months ago
- Transformer related optimization, including BERT, GPT☆60Updated last year
- ☆124Updated 2 weeks ago
- A collection of memory efficient attention operators implemented in the Triton language.☆219Updated 5 months ago
- ☆64Updated 4 months ago
- A MoE impl for PyTorch, [ATC'23] SmartMoE☆57Updated last year
- FlagScale is a large model toolkit based on open-sourced projects.☆178Updated this week
- Transformer related optimization, including BERT, GPT☆39Updated last year
- ☆79Updated 2 months ago
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆76Updated 8 months ago
- ☆140Updated 7 months ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆29Updated 2 months ago
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆46Updated 3 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆57Updated 5 months ago
- Models and examples built with OneFlow☆96Updated last month
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆364Updated this week
- Train llm (bloom, llama, baichuan2-7b, chatglm3-6b) with deepspeed pipeline mode. Faster than zero/zero++/fsdp.☆90Updated 9 months ago
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated 11 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆53Updated 3 weeks ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆83Updated 6 months ago
- Transformer related optimization, including BERT, GPT☆17Updated last year
- [ACL 2024] A novel QAT with Self-Distillation framework to enhance ultra low-bit LLMs.☆85Updated 6 months ago
- A flexible and efficient training framework for large-scale alignment tasks☆211Updated this week
- Sequence-level 1F1B schedule for LLMs.☆17Updated 5 months ago
- Inferflow is an efficient and highly configurable inference engine for large language models (LLMs).☆236Updated 8 months ago
- ATC23 AE☆43Updated last year
- Materials for learning SGLang☆110Updated this week
- Simple Dynamic Batching Inference☆145Updated 2 years ago
- Zero Bubble Pipeline Parallelism☆283Updated last week
- ☆138Updated 2 weeks ago