最近的文章
caffe 源码阅读笔记
·30 字·1 分钟
caffe做部署是YYDS!
blob
layer
net
激活函数
卷积
reshape
slice
loss function
reduce
eltwise
argmax
一张图看懂分布式训练:DP、ZeRO、TP、PP、SP、CP 与 EP
·2506 字·6 分钟
最近学习 MIT 6.S191 Lecture 9,陆续把 DDP、ZeRO、FSDP2、Tensor Parallelism、Pipeline Parallelism、Sequence Parallelism、Context Parallelism,以及 Expert Parallelism 过了一遍。
Expert Parallelism:为什么让 Token 去找 Expert,而不是搬运参数?
·9000 字·18 分钟
上一篇把单个 MoE Layer 的执行流走了一遍:Router 选 Expert、Token Dispatch 按 Expert 重排、Grouped GEMM 统一调度、Combine 加权合回原始 Token 顺序。整条链路假设所有 Expert 都在同一张 GPU 上。
从 Dense FFN 到 MoE:Router、Token Dispatch 与 Grouped GEMM
·5127 字·11 分钟
上一篇 MoE 笔记把架构家族理了一遍——稀疏、稠密、Soft MoE,gate 在前还是在后,决定了算力形态和适用场景。
但那篇有一段伪代码一直让我不太舒服:
1for prob, idx in zip(topk_probs.unbind(1), topk_idx.unbind(1)): 2 out += prob.unsqueeze(1) * self.experts[idx](x) 逻辑上没问题。但想象一下真实场景:一个 batch 有几千个 token,每个 token 选 2 个 expert,就是上万个 token-expert 计算任务。用 Python for loop 逐个调 expert 的 forward,GPU 利用率会低得离谱。
从 OBI、OFI 到 Microprice:高频 Alpha 信号的直觉与验证
·5497 字·11 分钟
前两篇写 Bid-Ask Spread 和 Adverse Selection 以及执行感知回测的时候,讨论的核心问题是"预测对了方向,为什么交易还可能亏钱"。结论是执行成本、成交概率和 Adverse Selection 都会侵蚀预测收益。
Sequence Parallelism 与 Context Parallelism:为什么沿序列切分还不够?
·5188 字·11 分钟
上一篇 TP 写完之后,我对 Megatron 的 Column Parallel + Row Parallel 已经比较清楚了:前一层按输出特征切分,后一层按输入特征切分,两层之间不需要 AllGather,只在最后做一次 AllReduce 把部分和加起来。
为什么 Self-Attention 不能偷看未来?从 Causal Mask 到自回归训练
·4700 字·10 分钟
最近在看 MIT 6.S191 Lecture 9 讲 Context Parallelism 的时候,碰到了一个词:Causal Attention。
我对 Self-Attention 的公式并不陌生——之前那篇已经把 QKV 和 routing score table 拆得比较清楚了。但 Causal Attention 这个概念我一直没有认真想过。回头看,我写 Attention 公式、写 Multi-Head、写 FlashAttention 的时候,脑子里跑的都是"所有 token 都能看到所有 token"的 Full Self-Attention。
Norm 到底归一化了谁?从 BatchNorm、LayerNorm 到 RMSNorm
·5020 字·11 分钟
起因 # 最近在过 MIT 6.S191 和分布式训练相关的材料,各种 Norm 反复出现。LayerNorm 在写 Transformer Block 那篇时已经接触过,BatchNorm 在做 CV 的那几年天天调,RMSNorm 在看 LLM 代码时到处都是。单独看每个都不算陌生。
预测正确,策略为何仍可能亏钱?——从 Markout 到执行感知回测
·5564 字·12 分钟
上一篇写到,预测对价格方向,不等于交易能赚钱。主动成交要付执行成本,挂单等成交又可能筛出对自己不利的行情。
顺着这个问题往下想,我还有一个疑问:如果把手续费算进去,回测里也只让一部分挂单成交,是不是就离真实收益很近了?
为什么预测对了价格方向,交易却可能亏钱?——从 Bid/Ask 到 Adverse Selection
·5810 字·12 分钟
之前写 order book 和价格形成那篇时,花了不少篇幅搞清楚 bid/ask、maker/taker、以及成交量和价格之间的关系。写完之后我觉得这些概念已经理顺了,但最近在看加密货币量化执行的材料时,碰到了一个更尖锐的问题: