RulinShao / LightSeq
Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers
β204Updated 5 months ago
Alternatives and similar repositories for LightSeq:
Users that are interested in LightSeq are comparing it to the libraries listed below
- β217Updated 8 months ago
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inferenceβ235Updated 2 months ago
- π Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.β186Updated last week
- Triton-based implementation of Sparse Mixture of Experts.β194Updated 2 months ago
- PyTorch bindings for CUTLASS grouped GEMM.β86Updated 3 weeks ago
- β97Updated 5 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inferenceβ415Updated 3 weeks ago
- Zero Bubble Pipeline Parallelismβ317Updated 2 months ago
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Servingβ291Updated 6 months ago
- REST: Retrieval-Based Speculative Decoding, NAACL 2024β190Updated last month
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLMβ152Updated 6 months ago
- β72Updated 2 years ago
- [NeurIPS 2024] KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantizationβ327Updated 5 months ago
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decodingβ105Updated last month
- Spec-Bench: A Comprehensive Benchmark and Unified Evaluation Platform for Speculative Decoding (ACL 2024 Findings)β218Updated 3 months ago
- PyTorch bindings for CUTLASS grouped GEMM.β61Updated 2 months ago
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsityβ195Updated last year
- Explorations into some recent techniques surrounding speculative decodingβ233Updated last month
- Code associated with the paper **Draft & Verify: Lossless Large Language Model Acceleration via Self-Speculative Decoding**β153Updated 8 months ago
- [NeurIPS'23] H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Models.β419Updated 5 months ago
- Odysseus: Playground of LLM Sequence Parallelismβ64Updated 7 months ago
- β311Updated 9 months ago
- Ring attention implementation with flash attentionβ660Updated last month
- [ICML 2024] KIVI: A Tuning-Free Asymmetric 2bit Quantization for KV Cacheβ269Updated last week
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).β232Updated 3 months ago
- Applied AI experiments and examples for PyTorchβ215Updated last week
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)β84Updated 3 months ago
- Dynamic Memory Management for Serving LLMs without PagedAttentionβ273Updated last month
- π Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flashβ¦β216Updated this week
- 16-fold memory access reduction with nearly no lossβ72Updated 2 months ago