面向大模型分布式训练与推理的深度学习优化库,提升效率并降低使用门槛。
4.3万5048PythonApache-2.0更新于 前天
一本开源的大语言模型与多模态模型训练、微调及推理工程实践手册,附带大量脚本和命令。
这是一套关于大语言模型和多模态模型训练、微调及推理的开放方法论、工具与分步指南,内含大量可直接复制使用的脚本和命令,面向 LLM/VLM 训练工程师与运维人员。
内容基于作者训练 BLOOM-176B、IDEFICS-80B 等模型以及 RAG 模型的实践经验。全书按硬件(计算、存储、网络)、编排(包括 SLURM)、训练、推理、调试与测试等主题组织,并附有公开的训练日志合集。
仓库同时提供 PDF 和 EPUB 格式的电子书,以及两个可供 AI 智能体使用的 SKILL.md 文件,分别用于指导模型训练和 GPU 集群评估。
克隆代码仓库
git clone https://github.com/stas00/ml-engineering打开目录,阅读所需指南(本书为 Markdown 文件,无需安装)
概要由 AI 根据项目 README 生成;GitHub 数据每日更新。
面向大模型分布式训练与推理的深度学习优化库,提升效率并降低使用门槛。
基于 PyTorch 的大模型分布式训练与推理系统,提供并行策略和显存优化等能力。
PyTorch 原生的模型量化与优化库,覆盖训练和推理部署。
配套《从零构建大语言模型》一书的代码,用 PyTorch 分步实现类 GPT 的大语言模型。
微软出品的 12 周 24 课初学者 AI 课程,涵盖符号 AI、神经网络、计算机视觉、NLP 和 AI 伦理,配有 PyTorch 与 TensorFlow 代码。
用于可视化神经网络、深度学习和机器学习模型结构的查看器,支持 ONNX、PyTorch 等多种格式。