← 返回 AI 开源导航

Nanotron

训练与微调

一、项目简介

nanotron 是 Hugging Face 生态中的极简大规模训练库,用最少的代码实现 3D 并行(数据、张量、流水线),强调「可读性优先」,便于研究者理解与修改分布式训练核心逻辑。

二、核心特性

  • 极简代码:核心逻辑清晰易懂;
  • 3D 并行:数据 / 张量 / 流水线并行;
  • 可读性优先:适合学习与研究;
  • 与 HF 生态协同;
  • 支持常见 LLM 架构;
  • 便于二次开发。

三、适用场景

  • 学习分布式训练实现原理;
  • 研究 3D 并行策略;
  • 定制预训练/微调流程。

四、快速上手

git clone https://github.com/huggingface/nanotron
cd nanotron && pip install -e .
# 参考 examples 配置并行策略启动训练

五、注意事项

  • 偏研究/教学,生产需自行加固;
  • 集群环境配置复杂;
  • 注意版本与依赖兼容。