跳到正文
GitHub · 灵魂体仓库·· 19 天前AI 评分38

LLMForge:从零实现并训练 decoder-only Transformer 的实验性训练框架

acphotinakis/LLMForge

来源与原文

GitHub · 灵魂体仓库

发布时间:(北京时间)

阅读原始来源 · github.com ↗
完整链接与内容说明

https://github.com/acphotinakis/LLMForge

本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。

AI 导读

LLMForge 是一个用 PyTorch 从零实现并训练的 decoder-only Transformer 语言模型训练框架,涵盖模型架构、tokenization、数据加载、训练循环、checkpoint 管理与推理全流程。架构采用 RoPE、GQA、RMSNorm、SwiGLU 和因果自注意力,训练引擎支持 AdamW、梯度累积、学习率调度和可恢复 checkpoint。项目支持 PyTorch MPS 在 Apple Silicon 上运行,数据集使用 FineWeb-Edu 子集,目前为实验性研究项目,不分发数据集文件和模型权重。

来源:GitHub · 灵魂体仓库 · github.com