hao-ai-lab / cse234-w25-PALinks
☆41Updated 7 months ago
Alternatives and similar repositories for cse234-w25-PA
Users that are interested in cse234-w25-PA are comparing it to the libraries listed below
Sorting:
- ring-attention experiments☆155Updated last year
- Cataloging released Triton kernels.☆264Updated 2 months ago
- 🔥 LLM-powered GPU kernel synthesis: Train models to convert PyTorch ops into optimized Triton kernels via SFT+RL. Multi-turn compilation…☆94Updated last month
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆130Updated 11 months ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆151Updated 3 weeks ago
- 16-fold memory access reduction with nearly no loss☆106Updated 7 months ago
- Hydragen: High-Throughput LLM Inference with Shared Prefixes☆44Updated last year
- JAX backend for SGL☆146Updated this week
- [ICLR2025 Spotlight] MagicPIG: LSH Sampling for Efficient LLM Generation☆238Updated 10 months ago
- A minimal implementation of vllm.☆60Updated last year
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆169Updated last year
- Efficient LLM Inference over Long Sequences☆390Updated 4 months ago
- ☆246Updated this week
- ☆130Updated 5 months ago
- [ICLR 2025] Palu: Compressing KV-Cache with Low-Rank Projection☆144Updated 8 months ago
- Collection of kernels written in Triton language☆161Updated 7 months ago
- ☆215Updated 10 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆126Updated 5 months ago
- ☆97Updated 7 months ago
- [ICLR'25] Fast Inference of MoE Models with CPU-GPU Orchestration☆240Updated 11 months ago
- QuTLASS: CUTLASS-Powered Quantized BLAS for Deep Learning☆125Updated last week
- ☆43Updated last year
- a minimal cache manager for PagedAttention, on top of llama3.☆125Updated last year
- ☆93Updated last year
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆344Updated 3 months ago
- ☆146Updated 8 months ago
- KV cache compression for high-throughput LLM inference☆144Updated 9 months ago
- Allow torch tensor memory to be released and resumed later☆160Updated last week
- TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operators☆89Updated 4 months ago
- [NeurIPS 2024] Efficient LLM Scheduling by Learning to Rank☆62Updated last year