标签:PyTorch

技术标签,共 305 个项目。

transformers
huggingface·Python

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

预训练模型+4训练框架
168730+11835kv5.22.22026-11-10 10:57 UTC
stable-diffusion-webui
AUTOMATIC1111·Python

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

图像生成+4图像生成
165279+3331kv1.10.12026-03-02 07:00 UTC
ComfyUI
Comfy-Org·Python

基于节点图的扩散模型图形界面与后端,支持本地快速推理和 API 调用,方便搭建和调试图像生成工作流。

扩散模型+4图像生成
146089+60317kv0.44.02026-11-10 08:24 UTC
LLMs-from-scratch
rasbt·Jupyter Notebook

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

大语言模型+4零代码微调平台
108152+21217k2026-11-09 14:59 UTC
vllm
vllm-project·Python

面向大语言模型的高吞吐、低显存推理与服务引擎,通过 PagedAttention 等优化显著提升并发吞吐并降低显存占用,适合部署在线推理服务。

推理引擎+4大模型推理服务进阶
96670+17725kv0.33.0rc32026-11-10 11:13 UTC
annotated_deep_learning_paper_implementations
labmlai·Python

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

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

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

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

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

目标检测+4检测 / 分割
63270+8712kv8.4.2082026-11-10 09:31 UTC
Real-Time-Voice-Cloning
CorentinJ·Python

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

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

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

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

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

计算机视觉+4图像处理
51577+444.9k0.31.52026-11-10 11:15 UTC
Made-With-ML
GokuMohandas·Jupyter Notebook

一套从数据工程、模型训练到部署迭代的完整教程,帮你把机器学习项目做成可上线的生产级应用。

MLOps+4数据质量
49981+297.8kv1.1.02026-03-04 23:44 UTC
ray
ray-project·Python

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

分布式计算+4Pipeline 编排
44248+208.3kray-2.60.02026-11-10 10:22 UTC
DeepSpeed
deepspeedai·Python

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

分布式训练+4分布式训练
43316+95.1kv0.19.92026-11-10 11:01 UTC
VoxCPM
OpenBMB·Python

无需分词器的多语言语音合成模型,支持创意音色设计和逼真声音克隆,可直接从文本生成自然语音。

语音合成+4语音
42181+10374.8k2.0.32026-11-09 07:36 UTC
pytorch-image-models
huggingface·Python

收录最全的 PyTorch 图像骨干网络合集,涵盖 ResNet、ViT、ConvNeXt 等主流模型,自带训练、评估、推理和导出脚本及预训练权重,省去自己搭网络和找权重的麻烦。

图像分类+4检测 / 分割
37272+75.2kv1.0.312026-11-07 23:22 UTC
MockingBird
babysor·Python

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

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

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

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

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

扩散模型+4图像生成
34894+87.5kv0.42.12026-11-10 10:44 UTC
netron
lutzroeder·JavaScript

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

模型可视化+4格式转换
33672+83.2kv9.3.52026-11-09 12:35 UTC
mmdetection
open-mmlab·Python

基于 PyTorch 的目标检测工具箱与基准库,集成了 Faster R-CNN、DETR、Grounding DINO 等主流模型,方便训练、评测和复现检测与分割算法。

目标检测+4检测 / 分割
33032+69.8kv3.3.02024-08-21 02:01 UTC
fairseq
facebookresearch·Python

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

训练框架
322076.7k2025-09-30 11:16 UTC
pytorch-lightning
Lightning-AI·Python

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

深度学习训练+4训练框架
31447+13.9k2.6.62026-10-30 08:08 UTC
insightface
deepinsight·Python

一套开箱即用的 2D/3D 人脸分析算法库,集成 ArcFace、RetinaFace 等模型,覆盖人脸检测、对齐、识别和年龄估计,支持 PyTorch、MXNet 等多后端。

人脸识别+4检测 / 分割
30180+166.1kmodel-zoo2026-10-19 07:54 UTC
EasyOCR
JaidedAI·Python

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

光学字符识别+4图像处理
30133+33.6kv1.7.22025-12-05 12:23 UTC
fastai
fastai·Jupyter Notebook

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

深度学习+4训练框架
28264+27.6k2.8.122026-09-21 10:41 UTC
ultimatevocalremovergui
Anjok07·Python

基于深度神经网络的图形化人声分离工具,可把歌曲拆成人声和伴奏,方便做卡拉OK伴奏或二次混音。

人声分离+4音频处理
27074+602.1kv5.62026-10-06 22:39 UTC
deep-learning-for-image-processing
WZMIAOMIAO·Python

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

图像分类+4检测 / 分割
26449+28.2k2026-01-01 07:16 UTC
JARVIS
microsoft·Python

把大语言模型和 Hugging Face 上的各类模型连起来,让 LLM 自动挑选、调用合适的模型完成多模态任务,相当于给模型加了个调度中枢。

模型调度+4Agent 框架
25603+142.3k2025-07-29 13:44 UTC
pytorch-CycleGAN-and-pix2pix
junyanz·Python

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

图像翻译+4图像生成
25277+26.6k2025-08-06 17:58 UTC
pytorch_geometric
pyg-team·Python

基于 PyTorch 的图神经网络库,提供图卷积等 GNN 模型的构建与训练组件,省去自己实现图数据结构和消息传递的麻烦。

图神经网络+4开发库 / 框架
24139+14.1k2.8.12026-11-06 06:58 UTC
ncnn
Tencent·C++

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

推理框架+4端侧部署
24033+34.5k202610202026-11-10 10:48 UTC
best-of-ml-python
lukasmasuch

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

机器学习库+4分析 / BI
23860+23.1k2025.10.302026-11-05 14:09 UTC
onnxruntime
microsoft·C++

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

推理引擎+4格式转换
22275+154.4kv1.30.12026-11-10 11:19 UTC
datasets
huggingface·Python

Hugging Face 官方数据集库,汇集海量即用型 AI 数据集,并提供高效的数据加载与处理工具,省去自己找数据、写加载代码的麻烦。

数据集托管+4数据集托管
22123+63.6k5.1.02026-11-09 21:41 UTC
peft
huggingface·Python

用 LoRA、Adapter 等少量参数微调大模型和扩散模型,大幅降低显存与训练成本,是 Hugging Face 生态里做参数高效微调的标准库。

参数高效微调+4参数高效微调
21880+32.7kv0.21.22026-11-10 08:54 UTC
onnx
onnx·Python

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

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

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

金融大模型+4训练框架
21547+83.0kv1.0.02026-10-25 13:49 UTC
FunASR
modelscope·Python

开源语音识别工具包,覆盖训练、推理、流式 ASR、VAD、标点恢复和说话人分离全流程,并提供 OpenAI 兼容与 MCP 服务接口。

语音识别+4语音推理
20988+312.1kv1.4.242026-11-10 10:05 UTC
ml-engineering
stas00·Python

一本面向大模型工程实践的开源手册,系统讲解 GPU 训练、推理、调试、网络与扩展性等落地问题,帮工程师把模型真正跑起来。

大模型工程+4大模型推理服务
19690+101.3k2026-11-04 04:44 UTC
pytorch-deep-learning
mrdbourke·Jupyter Notebook

配套 PyTorch 深度学习课程的开源教程与代码,从零基础讲到能独立训练模型,适合边看边动手练。

PyTorch+4知识库
19392+115.2k2026-09-27 23:39 UTC
llama-cookbook
meta-llama·Jupyter Notebook

Llama 官方示例手册,手把手教你在推理、微调、RAG 等场景中落地 Llama 系列模型,并覆盖多家云服务商的端到端方案。

Llama+4零代码微调平台
18589+62.8kv0.0.52026-05-19 18:42 UTC
DocsGPT
arc53·Python

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

智能体构建+4RAG 框架
18367+42.3k1.1.02026-11-10 10:53 UTC
CodeFormer
sczhou·Python

基于码本查找 Transformer 的盲人脸修复模型,能在低质量、模糊或退化的人脸图像上恢复清晰细节,适合老照片修复和人脸增强场景。

人脸修复+4论文复现
18221+63.7kv0.1.02025-11-18 12:03 UTC
Awesome-pytorch-list
bharathgs

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

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

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

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

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

深度学习框架+4硬件加速适配
16264+141.1kv0.22.12026-11-10 08:05 UTC
kubeflow
kubeflow

基于 Kubernetes 的云原生机器学习平台,把训练、流水线编排、模型服务和 Notebook 开发统一到容器化集群里,解决 MLOps 各环节割裂、难以规模化调度的问题。

云原生+4Pipeline 编排
15956+12.7kredirect2026-09-30 23:20 UTC
stable-diffusion-webui-colab
camenduru·Jupyter Notebook

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

Stable Diffusion+4图像生成
158902.6k0.0.172025-12-16 05:49 UTC
TensorRT-LLM
NVIDIA·Python

用 Python API 定义大模型,并借助 TensorRT 在 NVIDIA GPU 上做极致优化推理,还提供 Python/C++ 运行时来编排高性能服务。

推理引擎+4大模型推理服务
15051+172.9kv1.4.0rc32026-11-10 10:47 UTC
machine-learning-zoomcamp
DataTalksClub·Jupyter Notebook

一套为期四个月的免费机器学习工程课程,覆盖从模型训练到 Docker、FastAPI、Kubernetes 部署的完整链路,适合想补齐工程落地能力的算法同学。

机器学习工程+4容器 / Kubernetes
14941+93.3kdl-models2026-10-26 06:41 UTC
OpenMythos
kyegomez·Python

从公开研究文献出发,用第一性原理复现 Claude Mythos 架构的理论实现,供研究者拆解其注意力机制与模型设计思路。

模型架构复现+3编码 Agent
14923+13.3k2026-05-23 11:58 UTC
RWKV-LM
BlinkDL·Python

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

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

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

分布式训练
146742.2k2026-09-17 09:15 UTC
flair
flairNLP·Python

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

序列标注+4训练框架
14398+12.1kv0.15.12026-10-19 11:03 UTC
open_clip
mlfoundations·Python

开源实现的 CLIP 多模态模型,提供图像-文本对比学习训练与预训练权重,支持零样本图像分类和跨模态检索。

多模态嵌入+4多模态嵌入
14252+41.3kv4.2.12026-11-06 02:48 UTC
Open3D
isl-org·C++

Open3D 是一个现代 3D 数据处理库,提供点云、网格、RGB-D 和 3D 可视化等算法与工具,支持 C++ 和 Python,可用于 3D 感知、重建与机器学习研究。

3D数据处理+43D 重建
14191+102.6kv0.20.02026-11-05 18:33 UTC
ivy
unifyai·Python

把 PyTorch、TensorFlow、JAX、NumPy 等框架的机器学习代码互相转换,省去为不同框架重写模型的麻烦。

框架转换+3开发库 / 框架
141665.4k1.0.0.52026-11-10 06:28 UTC
stable-baselines3
DLR-RM·Python

基于 PyTorch 的强化学习算法库,提供 PPO、SAC、TD3 等常用算法的可靠实现,开箱即用,适合机器人控制和仿真环境训练。

强化学习+4训练框架
13999+122.2kv2.9.22026-10-26 09:35 UTC
pytorch-grad-cam
jacobgil·Python

PyTorch 上的 Grad-CAM 系列可视化工具,支持 CNN 和 Vision Transformer,可用于分类、检测、分割、图像相似度等任务的模型可解释性分析。

可解释性+4检测 / 分割
130091.7k2026-11-01 05:07 UTC

共 305 个,这里展示前 60 个。