NVIDIA / Star-Attention
Efficient LLM Inference over Long Sequences
☆366Updated last month
Alternatives and similar repositories for Star-Attention:
Users that are interested in Star-Attention are comparing it to the libraries listed below
- LLM KV cache compression made easy☆452Updated 3 weeks ago
- [ICLR 2025] DuoAttention: Efficient Long-Context LLM Inference with Retrieval and Streaming Heads☆447Updated 2 months ago
- [MLSys'25] QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Serving; [MLSys'25] LServe: Efficient Long-sequence LLM Se…☆635Updated last month
- A throughput-oriented high-performance serving framework for LLMs☆794Updated 6 months ago
- Advanced Quantization Algorithm for LLMs/VLMs.☆417Updated this week
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆238Updated this week
- [NeurIPS 2024] KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantization☆340Updated 8 months ago
- Memory layers use a trainable key-value lookup mechanism to add extra parameters to a model without increasing FLOPs. Conceptually, spars…☆313Updated 4 months ago
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆303Updated 9 months ago
- Ring attention implementation with flash attention☆734Updated this week
- 🐳 Efficient Triton implementations for "Native Sparse Attention: Hardware-Aligned and Natively Trainable Sparse Attention"☆615Updated 3 weeks ago
- Towards Economical Inference: Enabling DeepSeek's Multi-Head Latent Attention in Any Transformer-based LLMs☆158Updated this week
- Fast low-bit matmul kernels in Triton☆285Updated this week
- Applied AI experiments and examples for PyTorch☆256Updated 3 weeks ago
- Perplexity GPU Kernels☆185Updated this week
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆230Updated 2 months ago
- This repository contains the experimental PyTorch native float8 training UX☆222Updated 8 months ago
- 📰 Must-read papers on KV Cache Compression (constantly updating 🤗).☆367Updated last week
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆470Updated this week
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆529Updated last month
- Distributed Triton for Parallel Systems☆372Updated this week
- A curated collection of resources, tutorials, and best practices for learning and mastering NVIDIA CUTLASS☆156Updated 3 weeks ago
- ☆235Updated 11 months ago
- Q-GaLore: Quantized GaLore with INT4 Projection and Layer-Adaptive Low-Rank Gradients.☆195Updated 8 months ago
- A sparse attention kernel supporting mix sparse patterns☆186Updated 2 months ago
- ☆185Updated 6 months ago
- Dynamic Memory Management for Serving LLMs without PagedAttention☆345Updated 2 weeks ago
- ring-attention experiments☆129Updated 5 months ago
- [NeurIPS'24 Spotlight, ICLR'25] To speed up Long-context LLMs' inference, approximate and dynamic sparse calculate the attention, which r…☆965Updated this week
- [ICML 2024] KIVI: A Tuning-Free Asymmetric 2bit Quantization for KV Cache☆287Updated 2 months ago