跳到主要内容
SUN / 笔记

BLOG / 766 ARTICLES

技术笔记

计算机基础、AI、后端与客户端开发笔记

766 篇文章
19 分钟
AI

deepspeed推理和训练优化概述

伴随着 Transformer 结构的提出,后续 LLM 参数开始从亿级到百亿、乃至万亿级增长,与超高速增长到模型参数量相比, GPU 显存的增长实属有限,有数据显示每 18 个月仅翻了 1.7 倍。模型参数量的增长和硬件的显存的增长之间的差距越来越大,但从目前趋势看,超大的计算量和参数量是目前 LLM 训练和推理的必经之路。在千亿模型上训练就需要 1.5T

14 分钟
AI

ReAct & Plan-and-Execute

在构建 LLM Agent 系统时,选择合适的推理模式至关重要。本文将深入对比两种主流的 Agent 推理模式:ReAct(Reasoning and Acting)和 Plan-and-Execute,通过实战案例帮助你做出正确的技术选型。

5 分钟
AI

prompt-cache论文速读

在 多轮对话、DocQA 和 RAG 客服系统场景下中,提示词会包含重复的文本段落,如系统消息、提示模板和用于上下文的文档,也就是说常见文本段的注意力状态(也就是 kv 缓冲向量)\{(k_1,v_1), (k_2,v_2),..,(k_n,v_n)\} 是可以重复利用的。Prompt Cache 论文基于这一关键观察,设计了一个提示词模块提出来明确定义这些

4 分钟
AI

flashattention-3论文速读

flashattention-2 虽然减少了非矩阵乘法的计算量,提高了运行速度,但是其并未充分利用最新硬件的功能,导致在 H100 GPU 上的利用率仅为 35%,由此作者提出了 flashattention-3,提出了3 项优化策略来提升 Hopper GPU 上的注意力计算性能:

3 分钟
AI

MoE专家并行详解

在 MOE 模块中,Router 负责决定发送 token 到 指定 expert 中,在实际模型推理过程中,可能某些 experts 接收 tokens 过多,某些 experts 接收 tokens 过少,这种现象叫做 expert 负载不均衡。这种情况不仅脱离了 MoE 的设计初衷(术业有专攻),也影响了 MoE 计算效率(分布式训练推理中各卡通讯时的

页