LLM Mechanics 最小示例

LLM Mechanics 最小示例

这 12 个文件分别验证教程中的一个机制。它们不下载模型或数据,也不执行训练。

python3 -m venv .venv
source .venv/bin/activate
python -m pip install -r requirements.txt
python run_all.py

预期最后一行:

12/12 examples passed
文件验证内容
01_bpe.pybyte-level BPE merge 与 UTF-8 round-trip
02_decoder_only.pypre-norm decoder block 的输入输出 shape
03_causal_mask.py未来位置 attention 权重严格为零
04_rope.py旋转保持范数与相对位置性质
05_rmsnorm.py手写 RMSNorm 与 PyTorch 对齐
06_swiglu.pySwiGLU 门控与 ReLU² 的差异
07_prefill_decode.py多 token prefill 与单 query decode
08_kv_cache.pyKV cache 追加、显存和数值一致性
09_sampling.pygreedy、temperature 与 top-p
10_repetition_eos.pyrepetition penalty 与 EOS 停止
11_teacher_forcing.pyinput / target 的一位偏移
12_perplexity.pyperplexity 与 tokenization 口径限制