shreyansh26 / FlashAttention-PyTorch
Implementation of FlashAttention in PyTorch
☆123Updated last year
Related projects ⓘ
Alternatives and complementary repositories for FlashAttention-PyTorch
- ☆134Updated last year
- Get down and dirty with FlashAttention2.0 in pytorch, plug in and play no complex CUDA kernels☆98Updated last year
- Low-bit optimizers for PyTorch☆119Updated last year
- flash attention tutorial written in python, triton, cuda, cutlass☆209Updated 5 months ago
- ☆64Updated 3 months ago
- (Unofficial) PyTorch implementation of grouped-query attention (GQA) from "GQA: Training Generalized Multi-Query Transformer Models from …☆133Updated 6 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆362Updated this week
- A collection of memory efficient attention operators implemented in the Triton language.☆219Updated 5 months ago
- Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models☆184Updated 6 months ago
- ☆289Updated 7 months ago
- Code repo for the paper "LLM-QAT Data-Free Quantization Aware Training for Large Language Models"☆254Updated 2 months ago
- Transformer related optimization, including BERT, GPT☆60Updated last year
- An easy-to-use package for implementing SmoothQuant for LLMs☆83Updated 6 months ago
- The official code for paper "parallel speculative decoding with adaptive draft length."☆24Updated 2 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆68Updated 4 months ago
- [USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Paral…☆46Updated 3 months ago
- Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers☆195Updated 3 months ago
- The official implementation of the EMNLP 2023 paper LLM-FP4☆167Updated 11 months ago
- Ring attention implementation with flash attention☆588Updated last week
- QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.☆89Updated last month
- [NeurIPS'23] H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Models.☆391Updated 3 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
- [ACL 2024] A novel QAT with Self-Distillation framework to enhance ultra low-bit LLMs.☆84Updated 6 months ago
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆147Updated 4 months ago
- [ICML 2024] KIVI: A Tuning-Free Asymmetric 2bit Quantization for KV Cache☆241Updated last month
- ☆79Updated 2 months ago
- Spec-Bench: A Comprehensive Benchmark and Unified Evaluation Platform for Speculative Decoding (ACL 2024 Findings)☆190Updated 3 weeks ago
- Implementation of Speculative Sampling as described in "Accelerating Large Language Model Decoding with Speculative Sampling" by Deepmind☆82Updated 8 months ago
- This repository contains integer operators on GPUs for PyTorch.☆183Updated last year
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆209Updated 3 weeks ago