taehokim20 / LLMem
LLMem: GPU Memory Estimation for Fine-Tuning Pre-Trained LLMs
☆21Updated last year
Alternatives and similar repositories for LLMem
Users that are interested in LLMem are comparing it to the libraries listed below
Sorting:
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)☆103Updated last month
- A simple calculation for LLM MFU.☆38Updated 2 months ago
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆116Updated 5 months ago
- ☆75Updated 3 weeks ago
- Skywork-MoE: A Deep Dive into Training Techniques for Mixture-of-Experts Language Models☆132Updated 11 months ago
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated last year
- Odysseus: Playground of LLM Sequence Parallelism☆69Updated 11 months ago
- ☆84Updated last month
- Efficient, Flexible, and Highly Fault-Tolerant Model Service Management Based on SGLang☆51Updated 6 months ago
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆81Updated last month
- A MoE impl for PyTorch, [ATC'23] SmartMoE☆62Updated last year
- Boosting 4-bit inference kernels with 2:4 Sparsity☆73Updated 8 months ago
- [NeurIPS 2024] Fast Best-of-N Decoding via Speculative Rejection☆44Updated 6 months ago
- ☆70Updated last week
- Vocabulary Parallelism☆19Updated 2 months ago
- ☆20Updated last month
- A minimal implementation of vllm.☆40Updated 9 months ago
- ☆117Updated last year
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆161Updated 10 months ago
- Implementation of the paper: "Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention" from Google in pyTO…☆55Updated 3 weeks ago
- Cascade Speculative Drafting☆29Updated last year
- The official implementation of paper: SimLayerKV: A Simple Framework for Layer-Level KV Cache Reduction.☆45Updated 7 months ago
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆36Updated last month
- [ICML 2023] SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Models☆11Updated last year
- KV cache compression for high-throughput LLM inference☆127Updated 3 months ago
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆39Updated last year
- An innovative method expediting LLMs via streamlined semi-autoregressive generation and draft verification.☆25Updated last month
- Distributed IO-aware Attention algorithm☆20Updated 8 months ago
- ☆59Updated this week
- Triton implementation of Flash Attention2.0☆34Updated last year