AX 爱鲜报

nanochat

nanochat 是极简 LLM 训练实验框架,单 GPU 节点即可完成分词、预训练、微调、评估与推理全流程,用约 100 美元训练出 GPT-2 能力模型。

爱鲜报评分

0.0

0 人评分

59k

星数

否

中文

Python

主语言

是

活跃

57

贡献者

人工分析

EDITORIAL

推荐理由

Karpathy 出品的极简 LLM 训练框架,代码可读性极高,适合学习大模型全流程。仅需一个 --depth 参数即可自动配置所有超参,大幅降低实验门槛。社区活跃,有 GPT-2 速通排行榜,适合研究者与爱好者复现与改进。

适用场景

  • 学习 LLM 训练全流程
  • 低成本复现 GPT-2 能力模型
  • 单 GPU 节点上的 LLM 实验
  • 研究预训练与缩放定律
  • 教学与代码阅读

优点

  • 代码极简且可修改,适合学习
  • 单参数控制模型规模,自动配置超参
  • 覆盖分词、预训练、微调、评估、推理全流程
  • 成本低,约 48 美元即可训练 GPT-2 级别模型
  • 有活跃社区和速通排行榜

缺点

  • 需要 GPU 资源,显存不足需调参
  • 主要面向实验而非生产部署
  • 部分代码路径未全面测试(如 xpu、mps)
  • 文档以英文为主
上手难度 进阶

nanochat 简介

nanochat 是 Andrej Karpathy 开源的极简 LLM 训练实验框架,目标是让开发者用最低成本在单 GPU 节点上完成大语言模型的全流程训练。官方描述为「100 美元能买到的最好的 ChatGPT」。

核心特点

  • 全流程覆盖:包含分词、预训练、微调、评估和推理。
  • 极简可改:代码量小,结构清晰,适合阅读和魔改。
  • 单参数控制规模:只需设置 --depth(Transformer 层数),其他超参数(宽度、头数、学习率等)自动按最优方式计算。
  • 低成本:在 8×H100 节点上约 2 小时、花费约 48 美元即可训练出 GPT-2 能力级别的模型;使用竞价实例可低至约 15 美元。

快速开始

项目使用 uv 管理依赖:

uv sync --extra gpu    # CUDA 环境
uv sync --extra cpu    # CPU / MPS 环境
source .venv/bin/activate

复现并对话 GPT-2 级别模型:

bash runs/speedrun.sh
python -m scripts.chat_cli

适用人群

  • 想学习 LLM 训练全流程的开发者
  • 需要低成本实验环境的研究者
  • 对大模型原理感兴趣的学生和爱好者

注意事项

  • 需要 GPU 资源,显存不足时需调整 --device-batch-size 等参数。
  • 项目定位为实验框架,不建议直接用于生产部署。
  • 文档和社区讨论以英文为主。

评论

登录后才可评论与评分

0 条评论

  • 暂无评论,来写第一条吧。