kst179 / fused-attentionLinks
Fast and low-memory attention layer written in CUDA
☆20Updated 2 years ago
Alternatives and similar repositories for fused-attention
Users that are interested in fused-attention are comparing it to the libraries listed below
Sorting:
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆123Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆139Updated 7 months ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆44Updated 10 months ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆96Updated 3 months ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆113Updated last year
- Implement Flash Attention using Cute.☆100Updated last year
- ☆160Updated 2 years ago
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆276Updated 5 months ago
- QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.☆153Updated 4 months ago
- ☆100Updated last year
- ☆85Updated 11 months ago
- ☆115Updated last year
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆332Updated last year
- ☆255Updated last year
- An easy-to-use package for implementing SmoothQuant for LLMs☆110Updated 9 months ago
- This repository contains integer operators on GPUs for PyTorch.☆236Updated 2 years ago
- ☆164Updated last year
- A collection of memory efficient attention operators implemented in the Triton language.☆287Updated last year
- Fast Hadamard transform in CUDA, with a PyTorch interface☆271Updated 2 months ago
- ☆77Updated last year
- Boosting 4-bit inference kernels with 2:4 Sparsity☆90Updated last year
- Framework to reduce autotune overhead to zero for well known deployments.☆92Updated 3 months ago
- llama INT4 cuda inference with AWQ☆55Updated 11 months ago
- ☆104Updated last year
- QuTLASS: CUTLASS-Powered Quantized BLAS for Deep Learning☆155Updated 2 months ago
- The official implementation of the EMNLP 2023 paper LLM-FP4☆219Updated 2 years ago
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆175Updated last year
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆191Updated 11 months ago
- Autonomous GPU Kernel Generation via Deep Agents☆202Updated this week
- ☆91Updated last month