探索 PyTorch 项目

筛选并发现收录的、配有高质量中文文档或教程的 GitHub 项目。

找到 208 个项目 | 数据更新于:2026-03-07 14:20
Horovod logo
14.7k
📝 Python

分布式深度学习训练框架,支持 TensorFlow、Keras、PyTorch 和 Apache MXNet。基于 MPI 模型,易于扩展,性能高效,可实现单 GPU 到多 GPU、多节点的无缝迁移。

PyTorch
TensorFlow
Spark
📦 6.9 MB
🕐 2025-12-02
已收藏!可在"我的账户"中查看
Burn logo
14.5k
📝 Rust

基于 Rust 的下一代张量库与深度学习框架,兼顾灵活性、效率与可移植性。支持多后端(CPU/GPU),提供自动微分与内核融合,优化数值计算与模型训练推理。

PyTorch
📦 70.5 MB
🕐 2026-03-07
已收藏!可在"我的账户"中查看
Rwkv Lm logo
14.4k
📝 Python

RWKV 是一种并行化 RNN,具备 Transformer 级别的 LLM 性能。RWKV-7 是线性时间、常数空间、无注意力的 RNN 架构,适用于 LLM 和多模态应用。支持高效推理和快速训练,社区资源丰富。

PyTorch
📦 30.1 MB
🕐 2026-03-05
已收藏!可在"我的账户"中查看
Flair logo
14.4k
📝 Python

一个简洁的框架,提供最先进的自然语言处理(NLP)功能。支持命名实体识别、情感分析、词性标注等任务,内置多种语言模型。基于 PyTorch 构建,易于训练自定义模型,并支持 Flair 嵌入和各类变换器。

PyTorch
📦 386.3 MB
🕐 2025-10-28
已收藏!可在"我的账户"中查看
Nni logo
14.3k
📝 Python

开源 AutoML 工具包,自动化机器学习生命周期,涵盖特征工程、神经架构搜索、模型压缩与超参数调优。支持 Python,提供丰富的算法与可视化界面。

PyTorch
TensorFlow
算法
📦 130.0 MB
🕐 2024-07-03
已收藏!可在"我的账户"中查看
Ivy logo
14.2k
📝 Python

在不同机器学习框架间转换代码,支持 PyTorch、TensorFlow、JAX 和 NumPy。提供 ivy.transpile 函数,简化模型迁移。

PyTorch
TensorFlow
📦 157.6 MB
🕐 2026-02-08
已收藏!可在"我的账户"中查看
Yolov7 logo
14.1k
📝 Jupyter Notebook

YOLOv7 是 YOLO 系列的最新实时目标检测器,基于可训练的免费包实现。它在 COCO 数据集上达到 51.4% AP,推理速度高达 161 fps。提供多种模型变体,支持 Docker 和 PyTorch 环境部署。

PyTorch
📦 74.2 MB
🕐 2024-08-19
已收藏!可在"我的账户"中查看
Open Clip logo
13.5k
📝 Python

OpenAI CLIP 的开源实现,支持多种模型架构与大规模数据集训练。提供预训练模型、零样本分类及易用 API。

PyTorch
📦 15.5 MB
🕐 2026-02-28
已收藏!可在"我的账户"中查看
Open3D logo
13.4k
📝 C++

开源 3D 数据处理库,支持 C++ 和 Python。提供 3D 数据结构与算法、场景重建、表面对齐、3D 可视化及物理渲染。集成 PyTorch/TensorFlow,支持 GPU 加速,适用于端到端 3D 工作流。

PyTorch
TensorFlow
数据可视化
📦 593.6 MB
🕐 2026-03-05
已收藏!可在"我的账户"中查看
Lora logo
13.3k
📝 Python

基于低秩分解的参数高效微调库,通过冻结原始权重并学习秩分解矩阵对,大幅减少可训练参数量,适用于 PyTorch 模型,支持 Hugging Face 集成,实现高效任务切换且不增加推理延迟。

PyTorch
📦 34.1 MB
🕐 2024-12-17
已收藏!可在"我的账户"中查看
Deep Learning With Tensorflow Book logo
13.3k
📝 Jupyter Notebook

基于 TensorFlow 2.0 的开源深度学习入门书,理论与实战结合,提供 PDF、代码及课件。被多所高校采用,获权威媒体报道。

PyTorch
TensorFlow
📦 125.3 MB
🕐 2021-08-30
已收藏!可在"我的账户"中查看
Tensorrt Llm logo
13.0k
📝 Python

提供易用的 Python API 定义大语言模型,支持前沿优化以在 NVIDIA GPU 上高效推理。包含 Python 和 C++ 运行时组件,实现高性能推理编排。

PyTorch
大语言模型
📦 1752.7 MB
🕐 2026-03-07
已收藏!可在"我的账户"中查看
Pytorch Book logo
12.8k
📝 Jupyter Notebook

PyTorch 入门与实战教程,涵盖基础使用、高级扩展及实战应用。包含神经网络、风格迁移、动漫生成、自动写诗等项目,提供完整代码与预训练模型。

PyTorch
📦 122.0 MB
🕐 2023-12-24
已收藏!可在"我的账户"中查看
Stable Baselines3 logo
12.8k
📝 Python

PyTorch 版本的可靠强化学习算法库,提供稳定实现、自定义环境与策略支持,包含详细文档和基准测试。

PyTorch
算法
📦 5.0 MB
🕐 2026-02-21
已收藏!可在"我的账户"中查看
Clip As Service logo
12.8k
📝 Python

提供基于 CLIP 的图像和文本嵌入服务,支持低延迟、高扩展性。具备快速推理、弹性伸缩、易用 API 及多协议支持,可无缝集成神经搜索生态。

PyTorch
📦 28.0 MB
🕐 2024-01-23
已收藏!可在"我的账户"中查看
Ai Learn logo
12.7k

提供近200个AI实战案例与项目,涵盖Python、数学、机器学习、深度学习、计算机视觉、自然语言处理等热门领域。包含免费配套教材,零基础入门,助力就业实战。

PyTorch
TensorFlow
Pandas
📦 1.3 MB
🕐 2024-06-02
已收藏!可在"我的账户"中查看
Machine Learning Zoomcamp logo
12.7k
📝 Jupyter Notebook

免费 4 个月机器学习工程课程,涵盖从 Python 建模到 Docker、FastAPI、Kubernetes 和 AWS Lambda 生产部署的全流程。提供直播班与自学模式,含作业、项目与证书。

PyTorch
TensorFlow
Docker
📦 10.8 MB
🕐 2025-12-27
已收藏!可在"我的账户"中查看
Pytorch Grad Cam logo
12.7k
📝 Python

PyTorch 高级 AI 可解释性库。支持 CNN、ViT、分类、目标检测、分割、图像相似度等。提供多种像素归因方法,如 GradCAM、GradCAM++、ScoreCAM 等。包含平滑方法与性能指标,适用于模型诊断与研究基准。

PyTorch
数据可视化
📦 137.5 MB
🕐 2025-04-07
已收藏!可在"我的账户"中查看
The Incredible Pytorch logo
12.5k

精心整理的 PyTorch 资源合集,涵盖教程、论文、项目、社区及各类应用,包括大语言模型、计算机视觉、自然语言处理、强化学习等前沿领域。

PyTorch
📦 473 KB
🕐 2025-11-26
已收藏!可在"我的账户"中查看
Chineseocr Lite logo
12.3k
📝 C++

超轻量级中文OCR,总模型仅4.7M,支持竖排文字识别。提供ncnn、mnn、tnn、onnxruntime等多种推理框架,支持C++、JVM、Android、.Net等多平台部署,部署简单。

PyTorch
📦 467.7 MB
🕐 2023-08-15
已收藏!可在"我的账户"中查看
External Attention Pytorch logo
12.2k
📝 Python

基于 PyTorch 的注意力机制、MLP、重参数化和卷积模块实现库,旨在帮助理解论文核心思想,提供即插即用的模块化组件。

PyTorch
📦 5.6 MB
🕐 2024-12-06
已收藏!可在"我的账户"中查看
Nano Vllm logo
12.1k
📝 Python

轻量级 vLLM 实现,从零构建。提供快速离线推理、可读代码库及优化套件,如前缀缓存、张量并行、Torch 编译和 CUDA 图。性能媲美 vLLM,代码约 1200 行 Python。

PyTorch
大语言模型
📦 435 KB
🕐 2025-11-04
已收藏!可在"我的账户"中查看
Allennlp logo
11.9k
📝 Python

基于 PyTorch 的开源 NLP 研究库,用于开发各类语言任务的最先进深度学习模型。现已进入维护模式,推荐迁移到 AI2 Tango 等替代方案。

PyTorch
📦 73.0 MB
🕐 2022-11-23
已收藏!可在"我的账户"中查看
Ludwig logo
11.7k
📝 Python

低代码框架,用于构建自定义 LLM、神经网络等 AI 模型。支持声明式 YAML 配置,具备分布式训练、参数高效微调、4 位量化等优化,可扩展至生产环境。

PyTorch
大语言模型
📦 29.7 MB
🕐 2026-03-03
已收藏!可在"我的账户"中查看
Transformers Tutorials logo
11.5k
📝 Jupyter Notebook

包含多个使用 HuggingFace Transformers 库的演示,涵盖音频、文本、图像处理,如 AST、BERT、BEiT、CLIPSeg 等模型,提供 PyTorch 实现和 Colab 链接。

PyTorch
📦 232.4 MB
🕐 2026-01-13
已收藏!可在"我的账户"中查看
Segmentation Modelstorch logo
11.4k
📝 Python

基于 PyTorch 的图像语义分割库,提供 500+ 预训练卷积和 Transformer 骨干网络。支持 12 种编码器-解码器架构,如 Unet 和 Segformer,API 简单易用,仅需两行代码即可创建神经网络。包含常用损失函数与指标,支持 ONNX 导出。

PyTorch
📦 17.1 MB
🕐 2025-12-23
已收藏!可在"我的账户"中查看
Speechbrain logo
11.3k
📝 Python

基于 PyTorch 的开源语音工具包,加速对话式 AI 开发。支持语音识别、说话人识别、语音增强等任务,提供 200+ 训练配方和 100+ 预训练模型,可轻松微调 Whisper、Wav2Vec2 等模型。

PyTorch
📦 101.9 MB
🕐 2026-03-02
已收藏!可在"我的账户"中查看
Nerfstudio logo
11.3k
📝 Python

为 NeRFs 提供协作友好的工作室,简化从创建到训练的端到端流程。模块化设计提升可解释性,支持社区贡献与扩展,提供教程和文档资源。

PyTorch
📦 119.1 MB
🕐 2025-07-29
已收藏!可在"我的账户"中查看
Pythonpark logo
11.3k
📝 Python

Python 开源自学编程项目,提供保姆级教程:AI实验室、数据结构、机器学习、深度学习、网络爬虫、大厂面经、资源分享。原创文章每周两篇,B站首发视频,欢迎加入交流群。

PyTorch
📦 2.8 MB
🕐 2024-11-26
已收藏!可在"我的账户"中查看
Pytorch Unet logo
11.2k
📝 Python

基于 PyTorch 的 U-Net 图像语义分割实现,支持高质量图像处理。提供 Docker 镜像与预训练模型,适用于多类分割、人像分割及医学影像分析。

PyTorch
📦 47.4 MB
🕐 2024-08-11
已收藏!可在"我的账户"中查看