fanshiqing / grouped_gemmLinks
PyTorch bindings for CUTLASS grouped GEMM.
☆135Updated last month
Alternatives and similar repositories for grouped_gemm
Users that are interested in grouped_gemm are comparing it to the libraries listed below
Sorting:
- PyTorch bindings for CUTLASS grouped GEMM.☆109Updated 2 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆277Updated last year
- Zero Bubble Pipeline Parallelism☆418Updated 3 months ago
- ☆145Updated 5 months ago
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆61Updated last year
- ☆97Updated 11 months ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆214Updated last year
- nnScaler: Compiling DNN models for Parallel Training☆115Updated this week
- Allow torch tensor memory to be released and resumed later☆109Updated last week
- Best practices for training DeepSeek, Mixtral, Qwen and other MoE models using Megatron Core.☆56Updated this week
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆261Updated last month
- ☆229Updated last year
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆217Updated last year
- Train speculative decoding models effortlessly and port them smoothly to SGLang serving.☆313Updated this week
- Utility scripts for PyTorch (e.g. Memory profiler that understands more low-level allocations such as NCCL)☆44Updated 2 weeks ago
- This repository contains integer operators on GPUs for PyTorch.☆213Updated last year
- Applied AI experiments and examples for PyTorch☆290Updated 2 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆549Updated last month
- Pipeline Parallelism Emulation and Visualization☆60Updated 2 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆104Updated 4 months ago
- A lightweight design for computation-communication overlap.☆155Updated last week
- ☆85Updated 3 years ago
- Sequence-level 1F1B schedule for LLMs.☆31Updated 2 months ago
- ☆106Updated 7 months ago
- ☆43Updated last year
- ☆332Updated last year
- flash attention tutorial written in python, triton, cuda, cutlass☆404Updated 3 months ago
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆227Updated 2 weeks ago
- A Quirky Assortment of CuTe Kernels☆407Updated this week
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆319Updated last year