RiseAI-Sys / DAXLinks
High performance inference engine for diffusion models
☆79Updated 3 weeks ago
Alternatives and similar repositories for DAX
Users that are interested in DAX are comparing it to the libraries listed below
Sorting:
- 🤗An Unified Cache Acceleration Toolbox for DiTs: Qwen-Image, Qwen-Image-Edit, FLUX.1, Wan2.1/2.2, etc.☆201Updated this week
- A parallelism VAE avoids OOM for high resolution image generation☆75Updated 3 weeks ago
- ☆173Updated 7 months ago
- [ICLR'25] ViDiT-Q: Efficient and Accurate Quantization of Diffusion Transformers for Image and Video Generation☆115Updated 5 months ago
- [ECCV24] MixDQ: Memory-Efficient Few-Step Text-to-Image Diffusion Models with Metric-Decoupled Mixed Precision Quantization☆44Updated 9 months ago
- FastCache: Fast Caching for Diffusion Transformer Through Learnable Linear Approximation [Efficient ML Model]☆32Updated last week
- A sparse attention kernel supporting mix sparse patterns☆281Updated 6 months ago
- 📚A curated list of Awesome Diffusion Inference Papers with Codes: Sampling, Cache, Quantization, Parallelism, etc.🎉☆377Updated last week
- To pioneer training long-context multi-modal transformer models☆44Updated 3 weeks ago
- An auxiliary project analysis of the characteristics of KV in DiT Attention.☆32Updated 9 months ago
- 🤖FFPA: Extend FlashAttention-2 with Split-D, ~O(1) SRAM complexity for large headdim, 1.8x~3x↑🎉 vs SDPA EA.☆211Updated 3 weeks ago
- [CVPR 2025] Q-DiT: Accurate Post-Training Quantization for Diffusion Transformers☆56Updated 11 months ago
- [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring☆219Updated last month
- Code implementation of GPTAQ (https://arxiv.org/abs/2504.02692)☆58Updated last month
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆100Updated 3 months ago
- QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.☆138Updated last week
- Patch convolution to avoid large GPU memory usage of Conv2D☆92Updated 7 months ago
- Code for Draft Attention☆90Updated 3 months ago
- 🎬 3.7× faster video generation E2E 🖼️ 1.6× faster image generation E2E ⚡ ColumnSparseAttn 9.3× vs FlashAttn‑3 💨 ColumnSparseGEMM 2.5× …☆82Updated 2 months ago
- ☆91Updated last week
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆231Updated 2 weeks ago
- Triton based sparse quantization attention kernel collection☆33Updated last week
- A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters☆47Updated last year
- [ICML 2025] Official PyTorch implementation of "FlatQuant: Flatness Matters for LLM Quantization"☆153Updated last month
- ☆81Updated 7 months ago
- ☆98Updated 11 months ago
- ☆61Updated 4 months ago
- ☆408Updated 2 weeks ago
- Efficient Attention Methods: A Comprehensive Survey☆75Updated last week
- Quantized Attention on GPU☆44Updated 9 months ago