标签:PyTorch

305 条
#仓库Stars
1huggingface/transformers
模型定义的公共层,把权重、配置和结构对齐,让训练框架和推理引擎都从同一份代码读模型
169k
2AUTOMATIC1111/stable-diffusion-webui
把 Stable Diffusion 从脚本变成可交互工作台,是本地出图流程的事实标准前端
165k
3Comfy-Org/ComfyUI
节点图式的扩散模型推理后端,把模型、采样、后处理串成可复用的 JSON 工作流
146k
4rasbt/LLMs-from-scratch
配套书的 GPT 从零实现代码,用来把 LLM 训练流程拆开看一遍,不是拿来训生产模型
108k
5vllm-project/vllm
单卡也能压榨出高吞吐的 LLM 推理引擎,PagedAttention 基本成了行业标准
97k
6labmlai/annotated_deep_learning_paper_implementations
把论文公式翻译成可逐行对照的 PyTorch 代码,用来读算法而不是直接上生产
68k
7keras-team/keras
一套模型代码在 JAX、TensorFlow、PyTorch 之间切换,用来替代被单一后端锁死的训练脚本
64k
8ultralytics/ultralytics
一个包覆盖检测/分割/姿态/跟踪全流程,替代自建训练脚本和分散的模型仓库
63k
9CorentinJ/Real-Time-Voice-Cloning
SV2TTS 三阶段语音克隆的参考实现,用来把几秒音频变成可复用的说话人嵌入
60k
10ultralytics/yolov5
PyTorch 生态里最省事的检测/分割基线,用来快速验证数据和导出部署模型
58k
11roboflow/supervision
把各框架的检测输出统一成 sv.Detections,让标注、切片推理和数据集转换不再各写一套
52k
12GokuMohandas/Made-With-ML
一套把 ML 从 notebook 推到 CI/CD 和线上服务的完整工程化教程,用来替换零散的 MLOps 拼装经验
50k
13ray-project/ray
把单机 Python 并行代码原样扩到集群的分布式运行时,顶替手写 multiprocessing 加调度层
44k
14deepspeedai/DeepSpeed
在有限显存里塞进超大模型的 ZeRO 分片与并行训练层,直接替换 PyTorch DDP 的显存瓶颈
43k
15OpenBMB/VoxCPM
无分词器的连续语音表示 TTS,用自然语言描述造音色、短音频克隆,绕开离散 token 那套流程
42k
16huggingface/pytorch-image-models
PyTorch 图像骨干网络与预训练权重的统一入口,顶替散落各处的各模型官方实现
37k
17babysor/MockingBird
SV2TTS 的中文复现版,用 5 秒参考音频克隆音色,顶替从零训练 TTS 的整条链路
37k
18NandhaKishorM/laya
把分类、打分、是非判断压成一次前向的决策层,替掉你手写的 prompt 加解析和阈值逻辑
35k
19huggingface/diffusers
PyTorch 扩散模型的通用加载与调度层,顶替各模型仓库自带的推理脚本
35k
20lutzroeder/netron
模型文件的可视化查看器,替代自己写脚本解析 ONNX/TFLite 图结构
34k
21open-mmlab/mmdetection
PyTorch 检测/分割的模块化基座,用来替换自己手写的 backbone-neck-head 胶水代码
33k
22facebookresearch/fairseq
FAIR 的序列建模参考实现库,在需要复现 wav2vec 2.0/MMS 或改 Transformer 结构时顶替自己搭训练框架
32k
23Lightning-AI/pytorch-lightning
把 PyTorch 训练循环的工程样板收进 Trainer,换掉你自己手写的 DDP/AMP/checkpoint 胶水代码
31k
24deepinsight/insightface
人脸检测与识别模型的事实标准实现,替换掉自研 RetinaFace/ArcFace 训练管线
30k
25JaidedAI/EasyOCR
几行 Python 就能跑的多语种 OCR,用来替掉 Tesseract 在非拉丁文字和自然场景图上识别效果差的问题
30k
26fastai/fastai
PyTorch 上的高层训练封装,把视觉、文本、表格任务的训练循环压到几行代码
28k
27Anjok07/ultimatevocalremovergui
拖入音频就出人声/伴奏分轨的桌面工具,顶替掉手动跑 Demucs 脚本那一步
27k
28WZMIAOMIAO/deep-learning-for-image-processing
按网络逐个讲透结构并给可跑实现的图像处理教程仓库,用来替代零散博客和论文复现
26k
29microsoft/JARVIS
用 ChatGPT 做规划器、HuggingFace 模型做执行器的任务编排层,替代手写多模型调度逻辑
26k
30junyanz/pytorch-CycleGAN-and-pix2pix
PyTorch 里做无配对图像翻译的参考实现,顶替你自己从论文复现 CycleGAN 的那两周
25k
31pyg-team/pytorch_geometric
PyTorch 生态里做图神经网络的事实标准层,顶替你自己手写 message passing 和邻居聚合
24k
32Tencent/ncnn
不依赖 BLAS 的纯 C++ 推理运行时,用来把训练好的模型塞进 Android/iOS/嵌入式设备里跑
24k
33lukasmasuch/best-of-ml-python
选型前用来缩小候选集的一份按指标排序的 Python ML 库索引,不提供代码也不提供评测
24k
34microsoft/onnxruntime
把训练好的模型从框架里剥离出来,用统一运行时跑在 CPU/GPU/NPU 上,替代各框架自带推理引擎
22k
35huggingface/datasets
在训练脚本里替代手写 download+parse+split 胶水层,把 Hub 数据集和本地 CSV/Parquet 统一成同一个 Arrow 后端
22k
36huggingface/peft
在 Transformers 模型上注入 LoRA 等适配器,只训练千分之一参数并单独保存适配器权重
22k
37onnx/onnx
跨框架模型交换的中间表示层,导出链路的公共契约,不是推理引擎
22k
38AI4Finance-Foundation/FinGPT
金融领域 LoRA 微调与情感分析的参考实现,替代通用 LLM 在金融语料上的手工 prompt 调优
22k
39modelscope/FunASR
把 VAD、ASR、标点、说话人分离串成一条流水线的中文语音识别工具包,顶替自己拼 Whisper + pyannote 的胶水层
21k
40stas00/ml-engineering
一本讲大规模训练与推理运维细节的开源手册,顶替的是团队内部口口相传的排障经验和集群验收清单
20k
41mrdbourke/pytorch-deep-learning
从零学 PyTorch 的课程式知识库,顶替零散博客和官方 tutorial 拼凑出的学习路径
19k
42meta-llama/llama-cookbook
Meta 官方 Llama 示例库,用 notebook 顶替你自己从零搭的推理、微调、RAG 样板代码
19k
43arc53/DocsGPT
自托管的文档问答与智能体平台,顶替 SaaS 版 ChatGPT 企业知识库
18k
44sczhou/CodeFormer
盲脸修复的参考实现,用来替换 GFPGAN/GPEN 这类老一代人脸复原模型
18k
45bharathgs/Awesome-pytorch-list
一份 PyTorch 生态的索引清单,用来替代你自己在 GitHub 上翻找模型实现和辅助库
17k
46GeeeekExplorer/nano-vllm
用 1200 行 Python 复刻 vLLM 的调度与 KV 缓存,用来读源码而不是上生产
16k
47tracel-ai/burn
用 Rust 写训练代码,同一份代码直接跑推理,省掉导出模型这一步
16k
48kubeflow/kubeflow
在 K8s 上把训练、Notebook、推理串成一套可组合的 ML 平台,替代自建拼装方案
16k
49camenduru/stable-diffusion-webui-colab
在免费 Colab GPU 上跑起 Stable Diffusion WebUI 的笔记本集合,省掉本地显卡和装环境
16k
50NVIDIA/TensorRT-LLM
NVIDIA 官方 LLM 推理引擎,用预编译 engine 和专用 kernel 替代通用 PyTorch 推理路径
15k