CS336 Assignment1 训练组件：为手搓 Transformer 搭建训练与推理模块

斯坦福 CS336 课程笔记在从零搭建的 Transformer 模型基础上，补齐模型训练所需的基本模块，并用这些模块完成训练与推理验证。内容承接此前分词器与 Transformer 搭建部分，属于该课程作业的后续实践环节。