chengzeyi / piflux
(WIP) Parallel inference for black-forest-labs' FLUX model.
☆11Updated this week
Related projects ⓘ
Alternatives and complementary repositories for piflux
- [WIP] Context parallel attention that works with torch.compile☆49Updated this week
- A parallelism VAE avoids OOM for high resolution image generation☆40Updated last month
- Quantized Attention on GPU☆30Updated 2 weeks ago
- A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters☆32Updated 3 months ago
- faster parallel inference of mochi-1 video generation model☆73Updated this week
- ☆35Updated 2 weeks ago
- ☆46Updated 2 months ago
- Patch convolution to avoid large GPU memory usage of Conv2D☆79Updated 5 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆57Updated 5 months ago
- A CUDA kernel for NHWC GroupNorm for PyTorch☆15Updated this week
- ☆32Updated this week
- GPTQ inference TVM kernel☆36Updated 6 months ago
- Writing FLUX in Triton☆30Updated last month
- ☆100Updated last month
- IntLLaMA: A fast and light quantization solution for LLaMA☆18Updated last year
- ☆43Updated 11 months ago
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆34Updated 8 months ago
- An algorithm for static activation quantization of LLMs☆77Updated last week
- Efficient, Flexible and Portable Structured Generation☆40Updated this week
- Open deep learning compiler stack for cpu, gpu and specialized accelerators☆15Updated this week
- PyTorch bindings for CUTLASS grouped GEMM.☆53Updated 3 weeks ago
- Debug print operator for cudagraph debugging☆10Updated 3 months ago
- Decoding Attention is specially optimized for multi head attention (MHA) using CUDA core for the decoding stage of LLM inference.☆23Updated 2 weeks ago
- Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆78Updated this week
- QuIP quantization☆46Updated 8 months ago
- OneFlow Serving☆20Updated 9 months ago
- ☆29Updated 5 months ago
- Official implementation of the ICLR 2024 paper AffineQuant☆22Updated 7 months ago
- ☆16Updated this week
- [ECCV24] MixDQ: Memory-Efficient Few-Step Text-to-Image Diffusion Models with Metric-Decoupled Mixed Precision Quantization☆31Updated 2 months ago