一份 PyTorch 资源大全,把教程、论文、项目、社区等链接整理在一起,方便按图索骥找学习材料和参考实现。
PyTorch 官方的生成式 AI 模型训练平台,提供原生分布式训练组件和可复用的并行策略,帮你在多卡集群上高效训练大模型。
基于 PyTorch 的度量学习库,提供对比损失、难例挖掘、采样器和检索指标,用来训练和评估相似度搜索所需的嵌入向量。
把 PyTorch、TorchScript 和 FX 模型编译成 TensorRT 引擎,在 NVIDIA GPU 上做推理加速,省去手写 TensorRT 网络的麻烦。
基于 PyTorch 的点云深度学习训练框架,集成 PointNet、KPConv、MinkowskiEngine 等主流模型,方便在 S3DIS、ScanNet 上做分割等任务。
用 PyTorch 从零手写实现 65 道大厂 ML/AI 面试真题,配 Jupyter Notebook、自动评分器和 MCP AI 导师,适合系统刷题备战机器学习面试。
基于 PyTorch 的推荐模型训练框架,封装了 CTR 预估、生成式推荐等常用算法,开箱即用且方便扩展,支持昇腾 NPU 和 ONNX 导出。
基于 PyTorch 的知识蒸馏框架,无需写代码即可复现和对比 28 种顶会蒸馏方法,附带训练模型、日志和配置,方便做可复现的深度学习研究。
基于 PyTorch 的医学图像分割非官方实现,支持 2D 和 3D 数据,方便快速搭建和训练分割模型。
把 ONNX 模型直接转成 PyTorch 网络,方便在 PyTorch 里继续微调或调试,省去手工重搭结构的麻烦。
基于 PyTorch 的投机解码模型训练库,支持 EAGLE3、FSDP 等方案,帮你在 vLLM 等推理服务里加速大模型生成。
基于 Gradio 的 Stable Diffusion 网页界面,把文生图、图生图、局部重绘等能力做成可视化操作面板,不用写代码就能跑图像生成。
用 5 秒语音样本克隆音色,实时合成任意文本的语音,基于 PyTorch/TensorFlow 的说话人编码与声码器流水线。
基于 PyTorch 的扩散模型工具箱,统一加载和训练 Stable Diffusion、FLUX 等模型,做图像、视频、音频生成。
从公开研究文献出发,用第一性原理复现 Claude Mythos 架构的理论实现,供研究者拆解其注意力机制与模型设计思路。
基于 PyTorch 的开源语音工具包,覆盖语音识别、说话人识别与分离、语音增强等任务,提供预训练模型和训练流程,方便快速搭建语音系统。
让大模型在普通显卡上跑起来的 PyTorch 量化库,支持 8-bit 和 4-bit 权重压缩,也是 QLoRA 微调背后的核心依赖。
预训练语音合成模型集合,支持俄语、亚美尼亚语等多种语言,几行代码就能把文本转成语音,省去自己训练模型的麻烦。
一条命令就能搭好包含 Caffe、Keras、MXNet 等主流框架的深度学习 Docker 环境,省去逐个配置依赖的麻烦。
基于 ONNX Runtime、OpenVINO、PaddlePaddle 等多家推理后端的 OCR 工具包,支持多语言识别,方便在不同硬件上快速部署文字检测与识别。
把数据库和 AI 模型直接打通的全栈框架,让你在 MongoDB 等数据源上直接做推理、训练和部署自定义 AI 应用与智能体。
基于 C++ 的深度学习服务端与命令行工具,封装 Torch、TensorRT 等后端,通过 REST API 提供图像分类、目标检测、分割和以图搜图能力。
把 ONNX 模型一键转成 TFLite、TensorFlow、PyTorch 原生代码或 TorchScript,省去跨框架部署时手写转换脚本的麻烦。
为 Jetson Nano 准备的 Ubuntu 20.04 系统镜像,预装 CUDA、OpenCV、PyTorch、TensorFlow、TensorRT 等深度学习环境,省去在边缘设备上手动配置的麻烦。
基于 PyTorch/MXNet 的 Python 文字检测工具包,能从图片中定位中英文场景文字、数学公式和版面结构,适合 OCR 前处理与文档解析。
把 YOLOv5 模型部署到 TensorRT、LibTorch、ONNX Runtime、TVM、NCNN 等推理后端的运行时工具链,省去自己写适配代码。
纯 Rust 实现的大模型推理引擎,不依赖 PyTorch 和 Python 运行时,主打极快推理速度,适合部署 Qwen 等模型和构建 Agent 服务。
从零基础到 CV、NLP、LLM 项目实战再到 ONNX/TensorRT 工程化部署的 PyTorch 中文教程,配套 Jupyter Notebook 边学边练。
…onfiguration refactor ([ 2386](https://github.com/pytorch/torchtitan/pull/2386)) - External override registry ([ 3396](https://github.com/pytorch/torchtitan/pull/3396)) M
… on Blackwell by @rdspring1 in https://github.com/pytorch/ao/pull/4369 [nvfp4 training] Add Triton kernel for global amax of columnwise RHT (SM90+) by @rdspring1 in https
…commits since torch-geometric==2.6.0 . Highlights PyTorch 2.8 Support PyG 2.7 is fully compatible with PyTorch 2.8 and supports the following combinations: PyTorch 2.8 cp
…rhaul by @BloodAxe in https://github.com/BloodAxe/pytorch-toolbelt/pull/92 Feature/modules overhaul by @BloodAxe in https://github.com/BloodAxe/pytorch-toolbelt/pull/96 R
Hello everyone! We have just updated Pytorch-Wildlife to version 1.0.2.4. This is a small update but fixes some bugs related to result outputs reported in issue https://g