1 repositorio
GPU kernels that encode weights and activations in block-scaled low-precision formats such as FP4 and FP8, reducing memory bandwidth while preserving accuracy.
Distinct from Block-Wise Quantization: Distinct from Block-Wise Quantization: focuses on GPU kernel implementations for block-scaled formats (FP4, FP8) rather than general quantization strategies for optimizer states.
Explore 1 awesome GitHub repository matching artificial intelligence & ml · Block-Scaled Quantization Kernels. Refine with filters or upvote what's useful.
FlashInfer is a library of high-performance GPU kernels purpose-built for accelerating large language model inference. It provides optimized implementations for attention operations (including flash attention, page attention, multi-head latent attention, and cascade attention) using paged key-value caches, fused kernel composition, and just-in-time compilation. The library also includes specialized kernels for mixture-of-experts layers, block-scaled low-precision quantization (FP8, FP4), and distributed collective communication. What distinguishes FlashInfer is its fused all-reduce communicat
Provides GPU-accelerated block-scaled quantization kernels for FP4 and FP8 formats to reduce memory bandwidth in LLM inference.