训练框架

大规模预训练与微调的框架。

transformers
huggingface·Python

统一加载和运行文本、视觉、音频、多模态的预训练模型,覆盖推理与训练,省去为每个模型重写代码的麻烦。

预训练模型+4训练框架
168721+10935kv5.22.22026-11-10 03:52 UTC
stable-diffusion-webui
AUTOMATIC1111·Python

基于 Gradio 的 Stable Diffusion 网页界面,把文生图、图生图、局部重绘等能力做成可视化操作面板,不用写代码就能跑图像生成。

图像生成+4图像生成
165275+2931kv1.10.12026-03-02 07:00 UTC
LLMs-from-scratch
rasbt·Jupyter Notebook

从零用 PyTorch 一步步实现类 ChatGPT 大语言模型,覆盖注意力机制、预训练、指令微调全流程,适合边跑代码边理解 LLM 内部原理。

大语言模型+4零代码微调平台
108124+18417k2026-11-09 14:59 UTC
opencv
opencv·C++

开源计算机视觉库,提供图像处理、视频分析和深度学习推理的 C++ 接口,是视觉应用开发的基础设施。

计算机视觉+4图像处理
91508+3257k5.1.02026-11-10 03:19 UTC
tesseract
tesseract-ocr·C++

老牌开源 OCR 引擎,用 LSTM 识别图片中的文字,支持上百种语言,适合做文档扫描和文本提取。

OCR引擎+4文档解析 / OCR
77430+3111k5.5.42026-11-09 11:40 UTC
LlamaFactory
hiyouga·Python

一套统一接口搞定 100+ 大语言模型和多模态模型的微调,支持 LoRA 等高效训练方式,省去为每个模型单独写训练脚本的麻烦。

微调框架+4训练框架
75967+449.4kv0.9.52026-11-09 14:55 UTC
annotated_deep_learning_paper_implementations
labmlai·Python

用 PyTorch 逐行复现 60 多篇深度学习经典论文,代码旁配注释讲解,涵盖 Transformer、优化器、GAN、强化学习等,适合边读论文边对照实现学习。

论文复现+4参数高效微调
67741+276.8k2026-11-02 05:57 UTC
keras
keras-team·Python

面向人类友好的深度学习 API,用 Python 编写,可跑在 TensorFlow、JAX、PyTorch 上,帮开发者快速搭建和训练神经网络。

深度学习+4训练框架
64419+320kv3.15.12026-11-10 02:55 UTC
ultralytics
ultralytics·Python

YOLO 系列官方实现库,一套 PyTorch 代码覆盖目标检测、实例与语义分割、图像分类、姿态估计和目标跟踪,训练到部署全流程打通。

目标检测+4检测 / 分割
63256+7312kv8.4.2082026-11-10 04:19 UTC
Real-Time-Voice-Cloning
CorentinJ·Python

用 5 秒语音样本克隆音色,实时合成任意文本的语音,基于 PyTorch/TensorFlow 的说话人编码与声码器流水线。

语音克隆+4语音
60237+79.4k2026-03-09 10:31 UTC
yolov5
ultralytics·Python

基于 PyTorch 的目标检测全家桶,覆盖训练、推理、实例分割、图像分类和模型导出,开箱即用,适合快速落地视觉任务。

目标检测+4检测 / 分割
58257+417kv7.02026-11-07 16:59 UTC
supervision
roboflow·Python

一套可复用的计算机视觉工具库,提供目标检测、实例分割、分类等任务的标注处理、指标计算与可视化组件,帮开发者省去重复造轮子的时间。

计算机视觉+4图像处理
51571+384.9k0.31.52026-11-10 01:07 UTC
paperless-ngx
paperless-ngx·Python

把纸质文档扫描后自动 OCR、打标签并归档成可全文检索的个人文档管理系统,适合想彻底告别纸质文件柜的人。

文档管理+4文档解析 / OCR
47963+1713.4kv3.4.22026-11-10 00:31 UTC
ray
ray-project·Python

分布式 AI 计算引擎,提供核心分布式运行时和一组 AI 库,用来加速机器学习负载、大模型推理与超参搜索。

分布式计算+4Pipeline 编排
44245+178.3kray-2.60.02026-11-10 05:01 UTC
DeepSpeed
deepspeedai·Python

基于 PyTorch 的深度学习优化库,通过 ZeRO、模型并行和混合专家等技术,让千亿到万亿参数模型的分布式训练与推理更省显存、更高效。

分布式训练+4分布式训练
43315+85.1kv0.19.92026-11-10 01:13 UTC
ai-engineering-hub
patchy631·Jupyter Notebook

面向 LLM、RAG 和真实 AI Agent 应用的深度教程合集,用 Jupyter Notebook 带你从原理到落地跑通完整案例。

LLM教程+4RAG 框架
41657+3166.7k2026-11-01 04:39 UTC
langextract
google·Python

用 LLM 从非结构化文本中抽取结构化信息,并给出精确的原文出处定位和交互式可视化,方便核对抽取结果。

信息抽取+4训练框架
39297+232.7kv1.7.22026-11-10 01:39 UTC
tesseract.js
naptha·JavaScript

纯 JavaScript 实现的 OCR 库,基于 WebAssembly 在浏览器和 Node.js 中直接识别 100 多种语言的图片文字,无需后端服务。

OCR+4文档解析 / OCR
38827+82.4kv7.0.02026-05-17 04:19 UTC
mediapipe
google-ai-edge·C++

跨平台机器学习框架,提供可定制的视觉、音频等实时流媒体处理方案,方便在移动端和桌面端搭建低延迟推理管线。

端侧推理+4端侧部署
37624+356.2kv1.2.22026-11-10 04:08 UTC
MockingBird
babysor·Python

5 秒克隆任意人声,实时合成任意语音,基于 PyTorch 的语音克隆与文本转语音工具。

语音克隆+4语音
36873+25.2kv0.0.12026-03-03 14:59 UTC
laya
NandhaKishorM·Python

单次前向传播即可对任意文本做类型化选择、打分和是非判断的非自回归决策引擎,支持100多种语言,并按请求自动路由到合适的检查点。

决策引擎+4训练框架
35315+1333.2kv0.4.302026-11-09 06:53 UTC
diffusers
huggingface·Python

基于 PyTorch 的扩散模型工具箱,统一加载和训练 Stable Diffusion、FLUX 等模型,做图像、视频、音频生成。

扩散模型+4图像生成
34891+57.5kv0.42.12026-11-10 03:52 UTC
netron
lutzroeder·JavaScript

把 ONNX、PyTorch、Keras、CoreML 等格式的神经网络模型直接拖进浏览器就能看结构图,排查模型转换和层连接问题很方便。

模型可视化+4格式转换
33670+63.2kv9.3.52026-11-09 12:35 UTC
fairseq
facebookresearch·Python

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

训练框架
322076.7k2025-09-30 11:16 UTC
storm
stanford-oval·Python

输入一个主题,就能自动联网检索并生成带引用的长篇调研报告,省去人工查资料和整理成文的时间。

深度研究+4训练框架
31846+253.0kv1.1.02025-09-30 18:07 UTC
pytorch-lightning
Lightning-AI·Python

统一的 PyTorch 训练框架,让同一份代码在单卡到上万张 GPU 上都能跑,免去改代码就能做预训练和微调。

深度学习训练+4训练框架
314463.9k2.6.62026-10-30 08:08 UTC
EasyOCR
JaidedAI·Python

开箱即用的 OCR 工具,支持 80 多种语言和拉丁、中文、阿拉伯、天城文、西里尔等主流文字,几行 Python 代码就能把图片里的文字提取出来。

光学字符识别+4图像处理
30134+43.6kv1.7.22025-12-05 12:23 UTC
Resume-Matcher
srbhr·Python

本地运行的简历匹配与生成工具,用 100+ 大模型解析简历、计算与职位描述的文本相似度,并生成 PDF 和求职信。

简历解析+4向量数据库
29215+435.1kv1.4.02026-11-07 18:09 UTC
fastai
fastai·Jupyter Notebook

基于 PyTorch 的深度学习高层封装库,用少量代码就能训练出高精度模型,配套 notebook 教程降低入门门槛。

深度学习+4训练框架
28265+37.6k2.8.122026-09-21 10:41 UTC
DeepSpeech
mozilla·C++

DeepSpeech is an open source embedded (offline, on-device) speech-to-text engine which can run in real time on devices ranging from a Raspberry Pi 4 to high power GPU servers.

IoT / 固件
267724.1k2025-06-19 12:52 UTC
faster-whisper
SYSTRAN·Python

基于 CTranslate2 重实现的 Whisper 语音转文字推理引擎,通过量化与高效推理大幅提速并降低显存占用,适合本地部署转录。

语音识别+4语音推理
26633+612.2kv1.2.12026-11-06 17:24 UTC
deep-learning-for-image-processing
WZMIAOMIAO·Python

配套 B 站视频教程的深度学习图像处理实战代码库,用 PyTorch 和 TensorFlow2 覆盖图像分类、目标检测与分割的完整训练流程。

图像分类+4检测 / 分割
26449+28.2k2026-01-01 07:16 UTC
pytorch-CycleGAN-and-pix2pix
junyanz·Python

PyTorch 版 CycleGAN 与 pix2pix 实现,提供图像到图像翻译的完整训练与测试流程,支持无配对和有配对数据转换。

图像翻译+4图像生成
252756.6k2025-08-06 17:58 UTC
CV
AccumulateMore·Jupyter Notebook

汇集土堆 PyTorch、李沐《动手学深度学习》、吴恩达深度学习和大模型 Agent 四套课程的中文笔记,适合系统入门深度学习与大模型应用开发。

深度学习笔记+4训练框架
24266+232.7k2026-09-20 00:41 UTC
ncnn
Tencent·C++

面向手机端的高性能神经网络推理框架,针对 ARM 平台深度优化,让 Caffe、ONNX 等模型能在 Android/iOS 上低延迟运行。

推理框架+4端侧部署
24032+24.5k202610202026-11-10 04:45 UTC
best-of-ml-python
lukasmasuch

每周更新的机器学习 Python 库排行榜,按类别和评分整理优质库,帮你快速找到 AutoML、深度学习、数据可视化等方向的趁手工具。

机器学习库+4分析 / BI
23860+23.1k2025.10.302026-11-05 14:09 UTC
CVPR2026-Papers-with-Code
amusi

汇总 CVPR 2026 录用论文及其配套开源代码,方便计算机视觉研究者快速找到带实现的新工作并跟进复现。

计算机视觉+4图像处理
22951+52.8k2026-03-08 07:27 UTC
Awesome-Chinese-LLM
AiHubCN

汇总可私有化部署的中文大语言模型资源清单,覆盖小规模底座模型、垂直领域微调、数据集与教程,方便低成本选型落地。

中文大模型+4训练框架
22779+12.1k2026-05-10 05:03 UTC
screenpipe
screenpipe·Rust

持续录制电脑屏幕和音频,把日常工作流整理成可检索的记录,帮你找出值得自动化的环节,并为 AI 智能体提供上下文。

屏幕录制+4视觉语言模型
22317+582.3kapp-v2.7.1452026-11-10 00:40 UTC
onnxruntime
microsoft·C++

跨平台高性能推理与训练加速引擎,可直接加载 ONNX 模型,兼容 PyTorch、TensorFlow 等框架导出,解决模型在不同硬件上高效部署的问题。

推理引擎+4格式转换
22271+114.4kv1.30.12026-11-10 05:21 UTC
recommenders
recommenders-team·Python

微软出品的推荐系统实践库,提供从数据准备、模型训练到部署上线的完整示例和最佳实践,帮算法工程师快速搭建推荐与排序模型。

推荐系统+4容器 / Kubernetes
21985+43.3k1.2.12026-11-05 17:58 UTC
ai-guide
liyupi·JavaScript

程序员鱼皮整理的 AI 资源大全与零基础教程,涵盖大模型玩法、Prompt 提示词、AI 编程工具和开发框架,帮新手快速上手 AI 技术。

AI 教程+4RAG 框架
21980+422.4k2026-10-22 03:01 UTC
onnx
onnx·Python

ONNX 是开放的机器学习模型交换格式标准,让 PyTorch、TensorFlow、Keras 等框架训练出的模型能互相转换、跨平台部署。

模型交换格式+4格式转换
21668+54.1kv1.23.22026-11-09 23:01 UTC
FinGPT
AI4Finance-Foundation·Jupyter Notebook

开源金融大语言模型,提供训练好的模型和微调流程,用于金融情感分析、智能投顾等场景。

金融大模型+4训练框架
21545+63.0kv1.0.02026-10-25 13:49 UTC
awesome-production-machine-learning
EthicalML

一份精选的机器学习生产化开源工具清单,覆盖模型部署、监控、版本管理和规模化扩展,帮你快速找到 MLOps 各环节的趁手工具。

MLOps+4MLOps 平台
21086+72.6krelease-2026-11-012026-11-08 10:41 UTC
Auto-claude-code-research-in-sleep
wanshuiyin·Python

一套纯 Markdown 的轻量技能包,让 Claude Code、Codex 等 LLM 智能体在无人值守时自主完成机器学习研究:跨模型评审、想法发现与实验自动化。

自主研究+4MCP 生态
18951+1141.5kv0.4.282026-11-08 10:12 UTC
magika
google·Rust

用深度学习快速准确地识别文件真实类型,不依赖扩展名,支持 MIME 检测,Rust 实现,可嵌入各类应用做内容安全与文件分类。

文件类型检测+4格式转换
18784+81.2kcli/v1.1.02026-11-09 21:44 UTC
DocsGPT
arc53·Python

面向企业私有化部署的 AI 平台,可搭建智能体、助手和文档检索问答,支持多模型接入、深度研究与 API 连接。

智能体构建+4RAG 框架
18366+32.3k1.1.02026-11-09 22:33 UTC
Megatron-LM
NVIDIA·Python

NVIDIA 的大规模 Transformer 训练框架,支持张量并行、流水线并行等模型并行策略,用来在成百上千张 GPU 上高效预训练大语言模型。

大模型训练+4训练框架
18363+154.7kcore_v0.20.02026-11-10 05:20 UTC
Awesome-pytorch-list
bharathgs

汇总 GitHub 上 PyTorch 相关的模型、实现、辅助库和教程,帮你在一个清单里快速找到需要的深度学习资源。

PyTorch+4训练框架
16748+42.8k2026-09-22 06:52 UTC
nano-vllm
GeeeekExplorer·Python

用 PyTorch 从零实现的轻量级 vLLM,代码精简易读,适合学习大模型推理引擎的调度、KV Cache 和 PagedAttention 等核心机制。

推理引擎+4大模型推理服务
16379+372.8k2026-04-26 05:10 UTC
burn
tracel-ai·Rust

Rust 写的张量库和深度学习框架,兼顾灵活性与效率,支持自动微分、跨平台后端和内核融合,可训练和部署神经网络。

深度学习框架+4硬件加速适配
16264+141.1kv0.22.12026-11-10 05:03 UTC
ms-swift
modelscope·Python

一站式大模型训练框架,支持 600+ 纯文本大模型和 300+ 多模态大模型的 CPT、SFT、DPO、GRPO 等训练,可用 LoRA 等 PEFT 方式或全参数微调。

训练框架+4训练框架
16072+141.8kv4.6.02026-11-10 04:57 UTC
stable-diffusion-webui-colab
camenduru·Jupyter Notebook

在 Google Colab 上一键跑起 Stable Diffusion WebUI 的笔记本合集,省去本地显卡和环境配置,直接在线出图、图生图、训练 LoRA 和 DreamBooth。

Stable Diffusion+4图像生成
158912.6k0.0.172025-12-16 05:49 UTC
unstructured
Unstructured-IO·HTML

把 PDF、DOCX、图片等复杂文档解析成干净结构化数据,方便喂给语言模型做 RAG 和检索。

文档解析+4文档解析 / OCR
15707+111.4k0.27.372026-11-06 21:29 UTC
mlops-zoomcamp
DataTalksClub·Jupyter Notebook

DataTalks.Club 出品的免费 MLOps 实战课程,覆盖模型部署、监控和工作流编排,帮你把机器学习模型从实验推到生产环境。

MLOps课程+4MLOps 平台
15540+93.1k2026-10-15 13:49 UTC
albumentations
albumentations-team·Python

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

检测 / 分割
152981.7k2025-06-25 23:21 UTC
botpress
botpress·TypeScript

开源平台,用于构建和部署基于 GPT/LLM 的对话式 Agent,集成提示词编排、工具调用与多渠道发布,省去从零搭建对话机器人的重复工作。

Agent 框架+4训练框架
14988+22.3kv12.30.92026-11-09 20:44 UTC
RWKV-LM
BlinkDL·Python

RWKV 是一种兼具 RNN 与 Transformer 优势的语言模型架构,支持像 GPT 一样并行训练,推理时线性时间、常量显存、无限上下文,无需 KV 缓存。

语言模型+4训练框架
147721.0k5.002026-11-04 10:19 UTC
flair
flairNLP·Python

基于 PyTorch 的轻量级 NLP 框架,几行代码就能完成命名实体识别、词性标注、语义角色标注等序列标注任务的训练与微调。

序列标注+4训练框架
143972.1kv0.15.12026-10-19 11:03 UTC