xdit-project / xDiT
xDiT: A Scalable Inference Engine for Diffusion Transformers (DiTs) on multi-GPU Clusters
☆488Updated this week
Related projects: ⓘ
- [CVPR 2024 Highlight] DistriFusion: Distributed Parallel Inference for High-Resolution Diffusion Models☆554Updated last month
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆309Updated this week
- Ring attention implementation with flash attention☆529Updated this week
- 📖A small curated list of Awesome SD/DiT/ViT/Diffusion Inference with Distributed/Caching/Sampling: DistriFusion, PipeFusion, AsyncDiff, …☆64Updated 2 weeks ago
- ☆269Updated 5 months ago
- Analyze the inference of Large Language Models (LLMs). Analyze aspects like computation, storage, transmission, and hardware roofline mod…☆276Updated last week
- QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Serving☆399Updated 2 weeks ago
- Disaggregated serving system for Large Language Models (LLMs).☆278Updated last month
- [NeurIPS'23] H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Models.☆360Updated last month
- A collection of memory efficient attention operators implemented in the Triton language.☆205Updated 3 months ago
- This is the official PyTorch implementation of "LLMC: Benchmarking Large Language Model Quantization with a Versatile Compression Toolkit…☆227Updated this week
- Zero Bubble Pipeline Parallelism☆254Updated 2 weeks ago
- Microsoft Automatic Mixed Precision Library☆507Updated this week
- FlashInfer: Kernel Library for LLM Serving☆1,143Updated last week
- FlagGems is an operator library for large language models implemented in Triton Language.☆246Updated last week
- A throughput-oriented high-performance serving framework for LLMs☆470Updated this week
- Dynamic Memory Management for Serving LLMs without PagedAttention☆186Updated last month
- [ICCV 2023] Q-Diffusion: Quantizing Diffusion Models.☆313Updated 6 months ago
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆258Updated 2 months ago
- Official Implementation of EAGLE-1 and EAGLE-2☆749Updated 3 weeks ago
- Fast inference from large lauguage models via speculative decoding☆508Updated 3 weeks ago
- FP16xINT4 LLM inference kernel that can achieve near-ideal ~4x speedups up to medium batchsizes of 16-32 tokens.☆562Updated 2 weeks ago
- A fast communication-overlapping library for tensor parallelism on GPUs.☆184Updated this week
- A PyTorch Native LLM Training Framework☆581Updated 3 weeks ago
- 📰 Must-read papers and blogs on Speculative Decoding ⚡️☆350Updated last week
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆161Updated 2 months ago
- [CVPR 2024] DeepCache: Accelerating Diffusion Models for Free☆756Updated 2 months ago
- BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.☆342Updated this week
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆451Updated 6 months ago
- InternEvo is an open-sourced lightweight training framework aims to support model pre-training without the need for extensive dependencie…☆277Updated this week