ml-engineering

大模型训练推理踩坑指南,照着调优省半年时间

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 基础设施 🍴 复刻 1302 👀 关注 156
访问官网 ↗ GitHub ↗
GitHub 星标 ★ 19684
维护状态 活跃
是否开源 开源
定价模式 🏷️完全免费 开源书籍,CC-BY-SA-4.0许可,免费阅读,无付费版本。

项目预览

ml-engineering 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类基础设施
开发团队stas00
定价模式🏷️完全免费
价格说明开源书籍,CC-BY-SA-4.0许可,免费阅读,无付费版本。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议CC-BY-SA-4.0
主要语言Python
技术栈/模型ai,debugging,gpus,inference,large-language-models,llm,machine-learning,machine-learning-engineering,mlops,network,pytorch,scalability,slurm,storage,training,transformers
GitHub 星标★ 19684
上线时间2020-09-02 00:00:00
代码更新2026-11-04
仓库体积28.0 MB
本站数据更新2026-11-09
维护状态活跃
收录时间2026-08-09
浏览次数17

项目介绍

这是一本面向机器学习工程师的开放书籍,系统梳理了大模型训练与推理全流程的工程实践。它解决的核心问题是:当模型规模从百万级跃升到百亿级时,硬件选型、分布式策略、性能调优等工程细节如何落地。内容涵盖GPU/TPU硬件特性、网络拓扑(如NVLink、InfiniBand)、分布式训练框架(PyTorch/DeepSpeed/Megatron)、混合精度训练、梯度检查点、模型并行与数据并行组合、推理优化(如vLLM、TensorRT)等。书中包含大量真实案例的配置模板、故障排查清单和性能基准数据,例如如何通过调整batch size和通信协议将训练吞吐提升30%。它不同于纯理论教材,每个章节都直接对应生产环境中的具体操作,并附带可复现的代码片段。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 19684

使用教程

⚠ 本教程由 AI 依据该项目 README 整理;原始文档不完整,部分步骤为通用流程推断,实际安装请以项目官方文档为准。

难度:入门 约 5 分钟 部署方式:本地安装 4 步

⚠ 该项目 README 未提供完整安装说明,以下步骤为通用流程,请以官方文档为准。

环境要求

  • 可访问 GitHub 的网络环境
  • 本地已安装 git(用于克隆文档仓库)
  • 任意支持 Markdown 渲染的编辑器或浏览器(用于阅读章节)
  • 无需 Python 运行环境或额外依赖(README 未提供安装说明)

安装与启动步骤

  1. 1克隆文档仓库

    把这本书的全部 Markdown 章节下载到本地,README 本身就是入口目录。

    git clone https://github.com/stas00/ml-engineering.git
  2. 2进入仓库目录

    切换到克隆下来的目录,后续所有阅读和查找都基于此目录。

    cd ml-engineering
  3. 3查看章节结构

    列出目录内容,重点看 insights 等章节文件夹与 README 中的目录条目。

    ls -la
  4. 4按目录选读章节

    打开 README.md 的 Table of Contents,按兴趣进入对应章节文件阅读并复制示例命令。

如何确认成功

本地打开 README.md 能看到目录,并能进入目录中列出的章节文件,即表示获取成功。

常见问题

Q:这个项目需要 pip install 安装依赖吗?

A:不需要。README 没有给出任何安装命令,它是一本开放的工程文档书籍,克隆下来直接阅读 Markdown 文件即可。

Q:书里的命令能直接复制执行吗?

A:README 说明内容包含大量脚本和可直接复制粘贴的命令,用于快速解决训练与推理中的实际问题,但需自行核对硬件与框架版本。

Q:没有 GPU 或集群能看吗?

A:可以。内容面向 LLM/VLM 训练工程师和运维人员,涉及 GPU、网络、分布式训练等知识,先阅读再对照自己的环境实践。

Q:有在线版本吗?

A:项目提供了在线站点 https://stasosphere.com/machine-learning ,也可以直接在 GitHub 网页上阅读各章节。

注意事项

  • 该仓库是持续更新的经验笔记集合,内容会随作者实践不断增补。
  • 书中命令多为真实大规模训练场景(如 BLOOM-176B、IDEFICS-80B)的实践,执行前请确认与自身环境匹配。
  • README 未提供版本号、依赖清单或端口等部署信息,本教程仅为阅读与获取步骤。

核心亮点

  • 覆盖从硬件选型到分布式训练再到推理部署的完整链路,实操性强
  • 包含大量真实环境下的性能调优案例和配置模板,可直接复用
  • 持续更新,紧跟最新GPU架构和主流框架(如PyTorch 2.x)

不足之处

  • 内容偏重NVIDIA生态,对AMD/华为等硬件覆盖不足
  • 部分章节假设读者已有一定分布式训练基础,新手门槛较高

适用场景

  • 企业搭建大模型训练集群时进行硬件选型和网络规划
  • 训练过程中遇到显存不足或通信瓶颈,需要系统性排查调优
  • 模型上线前进行推理延迟优化和成本控制

替代项目

The Full Stack、Awesome LLM Engineering、Practical Deep Learning for Coders

同类项目推荐

unsloth 开源

本地一键跑起最新大模型,调教它不花一分钱

Local UI to run and train LLMs and diffusion models. Supports GGUF, MLX, Qwen3.8, De···

★ 79213 2026-08-09
MLOps 开源

跟着项目清单,动手练熟MLOps全流程

The repository contains a list of projects which I will work on while learning and i···

★ 79 2026-08-09
EasyR1 开源

多模态强化学习训练,高效扩展不卡壳

EasyR1: An Efficient, Scalable, Multi-Modality RL Training Framework based on veRL

★ 5211 2026-08-09
ml-engineering 开源

大模型训练推理踩坑指南,照着调优省半年时间

Machine Learning Engineering Open Book

★ 19684 2026-08-09
paradigms 开源

并行采样加速扩散模型,少步生成高质量图像

PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spo···

★ 159 2026-08-09
OneGen 开源

一次前向传播,边生成边检索,让 LLM 知识问答更快更准。

[EMNLP 2024] OneGen: Efficient One-Pass Unified Generation and Retrieval for LLMs.

★ 149 2026-08-09
DB-GPT-Hub 开源

一键微调大模型,让数据库查询更智能

A repository that contains models, datasets, and fine-tuning techniques for DB-GPT, ···

★ 2015 2026-08-09
how-to-train-your-gpt 开源

手把手从零写GPT,每行代码都讲透

Build a modern LLM from scratch. Every line commented. Explained like we are five.

★ 4175 2026-08-09
stable-pretraining 开源

一键搭建稳定可扩展的预训练流程,专注模型研究而非工程细节。

Reliable, minimal and scalable library for pretraining foundation and world models

★ 345 2026-08-09