AI Infra Academy
📚 AI Infra Wiki

系统源码与原理剖析

vLLM / SGLang / TensorRT-LLM / FlashAttention 等核心系统的深度技术文章。

全部推理引擎CUDAAttentionMoE训练综合

FlashAttention 为什么快:IO-aware 算法设计

标准 Attention 的瓶颈不在算力而在显存带宽。FlashAttention 通过 Tiling + 在线 softmax,把 HBM 访问从 O(N²) 降到 O(N²/M)。

FlashAttentionAttentionIO-awareCUDA
21 1 2