fanshiqing / grouped_gemm
PyTorch bindings for CUTLASS grouped GEMM.
☆67Updated 3 months ago
Related projects ⓘ
Alternatives and complementary repositories for grouped_gemm
- PyTorch bindings for CUTLASS grouped GEMM.☆51Updated last week
- A collection of memory efficient attention operators implemented in the Triton language.☆215Updated 5 months ago
- Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers☆196Updated 2 months ago
- ☆79Updated 2 months ago
- ☆63Updated 3 months ago
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆46Updated 3 months ago
- Zero Bubble Pipeline Parallelism☆279Updated this week
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆352Updated last week
- Odysseus: Playground of LLM Sequence Parallelism☆53Updated 4 months ago
- ☆70Updated 2 years ago
- ☆88Updated 2 months ago
- ☆121Updated this week
- ☆65Updated 3 years ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆26Updated 2 months ago
- ☆109Updated 7 months ago
- nnScaler: Compiling DNN models for Parallel Training☆62Updated 2 weeks ago
- Dynamic Memory Management for Serving LLMs without PagedAttention☆222Updated this week
- ☆74Updated 10 months ago
- Sequence-level 1F1B schedule for LLMs.☆17Updated 5 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆51Updated 2 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆82Updated 5 months ago
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆196Updated last week
- ☆162Updated 3 months ago
- This repository contains integer operators on GPUs for PyTorch.☆181Updated last year
- ☆183Updated 6 months ago
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆177Updated last year
- ☆140Updated 6 months ago
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆195Updated last week
- [ACL 2024] A novel QAT with Self-Distillation framework to enhance ultra low-bit LLMs.☆79Updated 5 months ago
- Materials for learning SGLang☆75Updated this week