SCENE COLLECTION

模型训练 方向开源项目精选

这里汇总了代码库收录的「模型训练」场景开源项目,按 Star 总量排序,适合按方向系统性地挑选与对比工具。

共收录 83 个项目 · 第 1 / 5 页

01

tensorflow

tensorflow / tensorflow

TensorFlow 是一个面向所有人的开源机器学习框架,支持从研究到生产的深度学习模型构建、训练与部署。

编辑点评作为最广泛使用的 ML 框架之一,TensorFlow 生态成熟,支持 Keras 高层 API、分布式训练和移动端部署,适合工业界与学术界。其静态图机制可能对新手有学习成本,但 TF 2.x 已大幅简化。注意版本碎片化问题,建议优先使用官方推荐版本。

模型训练 AI 基础设施 数据科学 深度学习 机器学习 神经网络 TensorFlow 模型训练
Apache-2.0 · 最近提交 2026-11-02
C++ 201k 78k ▲ -1
02

transformers

huggingface / transformers

预训练模型统一接口库,覆盖加载、微调与部署流程,支持数十万个开源模型。

编辑点评做 NLP 绕不开的库。几十万个预训练模型统一成一套接口,从加载、微调到最后部署都有现成路径。哪怕最后用别的框架,也建议先在这里找到合适的基座模型。

LLM 应用 模型训练 AI 基础设施 机器学习 预训练模型 基础设施
Apache-2.0 · 最近提交 2026-11-02
Python 168k 35k ▲ 15
03

ComfyUI

comfyanonymous / ComfyUI

节点式图像生成工作流引擎,把扩散模型的每一步拆成可连接的节点。

编辑点评图像生成玩到一定阶段,按钮式的 WebUI 就不够用了。ComfyUI 把每一步拆成节点,工作流能存成 JSON 复现和分享,批量出图也稳定。上手曲线陡,但天花板高。

AIGC 模型训练 Web 开发 AIGC 图像生成 节点式
GPL-3.0 · 最近提交 2026-11-02
Python 145k 17k ▲ 76
04

LLMs-from-scratch

rasbt / LLMs-from-scratch

Jupyter 开源项目,涉及AI、artificial-intelligence、attention-mechanism

编辑点评适用于AI/大模型、基础设施、数据科学场景。使用 Jupyter 开发,涉及AI、artificial-intelligence、attention-mechanism等技术方向。建议关注其社区活跃度和文档完善程度。

LLM 应用 模型训练 AI 基础设施 Jupyter Notebook AI artificial-intelligence attention-mechanism deep-learning finetuning
NOASSERTION · 最近提交 2026-10-30
Jupyter Notebook 108k 17k ▲ 25
05

pytorch

pytorch / pytorch

PyTorch 是一个基于 Python 的开源深度学习框架,提供张量计算与动态神经网络,并支持强大的 GPU 加速。

编辑点评PyTorch 凭借动态计算图和 Python 友好的接口,成为科研与工业界最主流的深度学习框架之一。它调试直观、生态丰富,适合快速原型验证和模型训练;但生产部署需搭配 TorchServe 或转 ONNX,且在高性能推理场景下需额外优化。

模型训练 数据科学 AI 基础设施 深度学习 GPU加速 张量计算 动态计算图 神经网络
NOASSERTION · 最近提交 2026-11-02
Python 105k 31k ▲ 10
06

vllm

vllm-project / vllm

高吞吐大模型推理引擎,用 PagedAttention 提升显存利用率,支持连续批处理与张量并行。

编辑点评自建推理服务时第一个该看的库。PagedAttention 把显存碎片问题处理得很干净,同样一张卡能撑更高并发。适合已经有模型、需要对外提供稳定 API 的团队。

LLM 应用 模型训练 AI 基础设施 大模型 推理加速 基础设施
Apache-2.0 · 最近提交 2026-11-02
Python 96k 24k ▲ 43
07

ML-For-Beginners

microsoft / ML-For-Beginners

微软出品的经典机器学习入门课程,12周26节课配52个测验,帮助初学者系统掌握ML基础。

编辑点评这是一份结构清晰、内容免费的机器学习课程,适合零基础或刚入门的学生自学。课程采用Jupyter Notebook实战教学,每节配套测验与练习,能有效巩固知识。微软官方维护,社区活跃,但内容偏基础,深度有限,需配合其他资料进阶。适合想快速建立ML知识框架的开发者。

数据科学 模型训练 开发工具 机器学习 初学者 Jupyter Notebook 微软 免费课程
MIT · 最近提交 2026-10-31
Jupyter Notebook 92k 23k ▲ 12
08

d2l-zh

d2l-ai / d2l-zh

《动手学深度学习》中文版,面向中文读者、可运行、能讨论的深度学习教材,被全球多所大学采用。

编辑点评这是中文开发者最推荐的深度学习入门教材之一,代码与理论结合紧密,支持Jupyter交互式运行,适合自学和教学。内容覆盖从基础到Transformer等前沿主题,但部分章节需要一定数学基础。建议搭配GPU环境实践,并积极参与社区讨论。

数据科学 模型训练 深度学习 动手实践 中文教材 PyTorch 机器学习
Apache-2.0 · 最近提交 2024-07-30
Python 82k 12k ▲ 9
09

ai-engineering-from-scratch

rohitg00 / ai-engineering-from-scratch

Python 开源项目,涉及agents、AI、AI-agents

编辑点评适用于AI/大模型场景。使用 Python 开发,涉及agents、AI、AI-agents等技术方向。建议关注其社区活跃度和文档完善程度。

LLM 应用 模型训练 Python agents AI AI-agents AI-engineering computer-vision
MIT · 最近提交 2026-11-02
Python 76k 14k ▲ 94
10

LlamaFactory

hiyouga / LlamaFactory

LlamaFactory 是一个统一的高效微调框架,支持 100+ 种大语言模型与视觉语言模型,帮助开发者快速完成模型定制。

编辑点评作为 ACL 2024 的入选项目,LlamaFactory 将 LoRA、QLoRA 等多种微调方法整合到统一接口中,大幅降低了 LLM/VLM 微调的上手门槛。适合需要快速实验或生产级适配的团队。注意:不同模型对硬件资源要求差异较大,建议先阅读官方文档确认显存与训练速度。

模型训练 开发工具 LLM 应用 大模型微调 高效训练 LLM VLMs 统一框架
Apache-2.0 · 最近提交 2026-10-31
Python 76k 9.3k ▲ 10
11

AI-For-Beginners

microsoft / AI-For-Beginners

微软出品的 AI 入门免费课程,12 周 24 课时,覆盖神经网络、NLP、计算机视觉等核心主题,适合零基础开发者系统学习。

编辑点评微软官方维护,内容结构清晰,每课配有 Jupyter Notebook 实战练习,理论与实践结合,非常适合想入门 AI 的开发者。课程覆盖面广但深度有限,适合作为学习路线图,后续需结合其他资源深入。全部免费开源,社区活跃,是极佳的自学起点。

数据科学 模型训练 开发工具 AI 入门 机器学习 深度学习 Jupyter Notebook 免费课程
MIT · 最近提交 2026-10-23
Jupyter Notebook 7k 14k ▲ 17
12

annotated_deep_learning_paper_implementations

labmlai / annotated_deep_learning_paper_implementations

60+深度学习论文实现与教程,带对照注释,涵盖Transformer、优化器、GAN、强化学习等,帮助理解与复现论文。

编辑点评每个实现都有配套笔记,适合想深入理解论文细节的开发者;代码结构清晰,覆盖范围广,从经典到前沿都有。注意:实现并非官方版本,需结合原文阅读;部分模型需要较深基础,适合有一定PyTorch经验的人。

模型训练 数据科学 AIGC 深度学习 论文实现 PyTorch Transformer 强化学习
MIT · 最近提交 2026-11-02
Python 68k 6.8k ▲ 7
13

keras

keras-team / keras

Keras 是一个为人类设计的深度学习 API,提供简洁、模块化的接口,帮助开发者快速构建和训练神经网络模型。

编辑点评Keras 以易用性著称,适合快速原型开发和教学,同时与 TensorFlow 深度集成,支持从研究到生产的全流程。新手友好,但高级自定义可能仍需深入底层,适合需要快速迭代和标准化建模的场景。

模型训练 数据科学 AI 基础设施 深度学习 神经网络 TensorFlow Python
Apache-2.0 · 最近提交 2026-11-01
Python 64k 2k ▲ 2
14

ultralytics

ultralytics / ultralytics

Ultralytics YOLO 官方库,提供目标检测、实例分割、姿态估计、分类与跟踪的一站式 PyTorch 实现。

编辑点评作为 YOLO 系列的官方仓库,它持续迭代并集成最新模型(如 YOLO11/YOLO26),API 简洁、训练推理部署闭环完善,适合快速落地视觉任务。但需注意依赖版本严格,且部分功能对新手有学习成本,推荐在明确需求后直接采用官方文档与 CLI 上手。

模型训练 数据科学 AI 基础设施 目标检测 实例分割 姿态估计 YOLO 计算机视觉
AGPL-3.0 · 最近提交 2026-11-02
Python 63k 12k ▲ 5
15

faceswap

deepfakes / faceswap

Deepfakes Faceswap 是一个开源的深度学习换脸工具,基于 Python 实现,让用户能够通过 AI 技术对视频和图片中的人脸进行替换与合成。

编辑点评作为深度学习换脸领域的经典开源项目,Faceswap 提供了完整的训练、转换和推理流程,适合研究 GAN、人脸生成和图像合成技术的开发者。项目代码结构清晰,支持多种模型和数据集,但需注意换脸技术可能涉及伦理与法律风险,使用时务必遵守相关法规并取得当事人同意,避免用于虚假信息传播或恶意用途。

AIGC 模型训练 数据科学 深度学习 人脸合成 换脸 GAN 开源工具
GPL-3.0 · 最近提交 2026-08-05
Python 58k 13k ▲ 2
16

qlib

microsoft / qlib

Qlib 是微软开源的 AI 量化投资平台,用机器学习赋能量化研究到生产全流程,支持监督学习、市场动态建模和强化学习等多种范式。

编辑点评Qlib 是量化投资领域少有的完整 AI 平台,覆盖数据、模型、回测到实盘全链路,内置丰富算法与数据集,适合量化研究员和 AI 工程师快速验证策略。其与 RD-Agent 集成可自动化研发流程,但学习曲线较陡,且需自行适配真实市场数据与交易接口。

数据科学 模型训练 AI 基础设施 量化投资 机器学习 金融科技 开源平台 回测框架
MIT · 最近提交 2026-10-30
Python 5k 7.9k ▲ 18
17

Made-With-ML

GokuMohandas / Made-With-ML

一个开源教程项目,系统讲解如何开发、部署并迭代生产级机器学习应用,帮助开发者构建可靠的ML系统。

编辑点评内容覆盖面广,从ML基础到MLOps实践都有涉及,尤其适合希望把模型真正落地到生产环境的开发者。项目以Jupyter Notebook为主,互动性强,但需要自己动手实践。注意部分内容可能依赖旧版库,建议结合最新官方文档使用。

AI 基础设施 数据科学 模型训练 机器学习 生产部署 MLOps Jupyter Notebook 学习教程
MIT · 最近提交 2026-03-04
Jupyter Notebook 5k 7.8k ▲ 14
18

ray

ray-project / ray

Ray 是一个 AI 计算引擎,包含分布式核心运行时和加速机器学习工作负载的 AI 库,用于分布式训练、调优和服务。

编辑点评Ray 是当前 AI 基础设施领域的事实标准,抽象了分布式调度的复杂性,让开发者可以用 Python 编写可扩展的并行应用。它特别适合大规模训练、超参调优、强化学习和在线推理等场景。缺点是学习曲线较陡,调试相对复杂,且部分高级功能依赖云环境。

AI 基础设施 模型训练 数据科学 分布式计算 机器学习 可扩展性
Apache-2.0 · 最近提交 2026-11-02
Python 44k 8.2k ▲ 0
19

TensorFlow-Examples

aymericdamien / TensorFlow-Examples

面向初学者的 TensorFlow 教程与示例集合,涵盖 TF v1 和 v2 的经典用法,帮助快速上手深度学习。

编辑点评这是学习 TensorFlow 的经典入门资源,示例覆盖基础操作、图像分类、自然语言处理等常见任务,且每个示例都配套完整代码和说明。支持 v1 与 v2 双版本,对维护旧代码或迁移到新版本的开发者尤其友好。优点是结构清晰、上手快;缺点是部分示例偏向基础,深度不足,适合初学者或需要快速参考的场景。

数据科学 模型训练 开发工具 TensorFlow 深度学习 入门教程 示例代码 机器学习
NOASSERTION · 最近提交 2024-07-26
Jupyter Notebook 44k 15k ▲ 1
20

DeepSpeed

deepspeedai / DeepSpeed

DeepSpeed 是微软开源的深度学习优化库,让分布式训练和推理更简单、高效,大幅降低大模型训练成本。

编辑点评DeepSpeed 是训练大规模 AI 模型的事实标准之一,其 ZeRO 显存优化技术能显著降低显存占用,支持千亿级参数训练。适合需要训练或微调大模型的研究团队和企业,同时提供推理优化、混合精度等能力。注意其配置较为复杂,上手门槛偏高,且部分优化依赖特定硬件和 CUDA 环境。

AI 基础设施 模型训练 AI Agent 分布式训练 大模型 显存优化 ZeRO 深度学习
Apache-2.0 · 最近提交 2026-11-02
Python 43k 5.1k ▲ 2

相关场景

继续浏览