AI Infra Academy
📚 AI Infra Wiki

系统源码与原理剖析

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

全部推理引擎CUDAAttentionMoE训练综合

MoE 模型与 Expert Parallelism

稀疏激活的 MoE 如何在保持算力可控的同时扩大模型容量,以及 Expert Parallelism 的通信与负载均衡挑战。

MoEExpert ParallelismDeepSeekDeepEP
8 1 0

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

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

FlashAttentionAttentionIO-awareCUDA
20 1 2

vLLM 与 PagedAttention 深度解析

从 KV Cache 显存碎片问题出发,剖析 vLLM 如何借鉴操作系统分页思想实现近乎零浪费的显存管理。

vLLMPagedAttentionKV Cache显存管理
12 1 0