跳到正文
GitHub · 去对齐工具与仓库·· 19 天前精选AI 评分59

AARUL:在笔记本 GPU 上从零训练的 15.7M 参数语言模型,可在浏览器中运行

AarulKoul/aarul-llm

来源与原文

GitHub · 去对齐工具与仓库

发布时间:(北京时间)

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

https://github.com/AarulKoul/aarul-llm

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

AI 导读

作者发布了完全从零编写的语言模型 AARUL:15.74M 参数的 decoder-only transformer,采用 Llama 同款设计,在 RTX 3070 Ti 笔记本 GPU 上用 TinyStories 数据训练 2.6 小时、229M tokens,验证损失从 8.40 降到 1.274。浏览器端以 19 KB 无依赖 JavaScript 运行 int8 量化模型(约 40-50 tokens/s),可实时展示 48 个注意力头的权重和各 token 概率分布,并通过与 PyTorch 对照的测试验证实现一致性。仓库 MIT 许可,含完整复现脚本。

推荐理由

作者从零实现训练与浏览器推理并用 PyTorch 浮点精度做一致性测试,展示了可复现的小模型全流程做法。

来源:GitHub · 去对齐工具与仓库 · github.com