PyTorch资源
探索与 PyTorch 相关的精选开源项目、速读教程和最新工作机会
相关开源项目
neuralchen/simswap
基于 PyTorch 的高效人脸交换框架,支持图像和视频的任意人脸替换,仅需单个训练模型即可实现高保真效果。提供训练与测试代码,支持高分辨率版本 SimSwap-HQ。
rasbt/reasoning-from-scratch
使用 PyTorch 从头实现推理大语言模型,逐步教程。教育代码库,涵盖推理时间缩放、强化学习、蒸馏,镜像大规模模型方法。
udacity/deep-reinforcement-learning
Udacity 深度强化学习 Nanodegree 课程配套仓库。包含 PyTorch 实现的多种强化学习算法教程,如 DQN、DDPG、PPO 等,以及基于 Unity ML-Agents 的导航、连续控制等实战项目。
nvidiagameworks/kaolin
基于 PyTorch 的 3D 深度学习加速库,提供模块化可微渲染、快速表示转换、数据加载、3D 检查点、可微相机与光照 API、结构化点云加速及 Jupyter 交互式可视化等 GPU 优化操作。
timesler/facenet-pytorch
基于 PyTorch 的人脸识别库,提供预训练的 MTCNN 人脸检测和 InceptionResnet 识别模型。支持快速部署与微调,包含完整检测识别流程示例。
open-mmlab/mmaction2
OpenMMLab 的下一代视频理解工具箱与基准,基于 Python,提供丰富的模型库和简洁的 API。
dmmaze/ballonstranslator
深度学习辅助漫画翻译,一键机翻支持日漫美漫,图像修复与文本编辑,LLM上下文术语表统一翻译,适用于条漫。
csailvision/semantic-segmentation-pytorch
基于 PyTorch 的 MIT ADE20K 数据集语义分割/场景解析实现。支持 PSPNet、UPerNet、HRNet 等 SOTA 模型,提供预训练模型与 Colab 演示。特色包括同步批归一化、多 GPU 动态输入尺度及高效训练。
tencentmusic/cube-studio
开源云原生一站式 AI 平台,支持 MLOps 全流程、分布式训练、超参搜索、推理服务 VGPU 虚拟化、大模型微调与多机推理,兼容国产硬件及多种框架。
autogptq/autogptq
基于 GPTQ 算法的易用 LLM 量化包,提供用户友好的 API。支持 int4 量化,提升推理速度并降低显存占用。已集成至 Transformers 等框架。