搜索结果：pytorch（55 个）
按分类缩小范围
多种注意力机制、MLP、重参数化及卷积的 PyTorch 实现，辅助理解学术论文。
cleanrl
综合 57
面向科研的高质量单文件深度强化学习算法实现（含 PPO、DQN、SAC 等）
基于 PyTorch 实现的图像到图像转换算法（CycleGAN 与 pix2pix）。
MockingBird
综合 54
仅需 5 秒即可克隆声音，并实时生成任意语音的开源工具。
CodeFormer
综合 53
NeurIPS 2022 成果，基于预设码本的鲁棒盲人脸修复与图像复原算法。
专注于图像处理的深度学习开源项目，涵盖图像分类、目标检测等核心领域与实践。
burn
综合 51
兼顾灵活性、高效性与跨平台特性的 Rust 深度学习框架。
功能强大的 Stable Diffusion 浏览器端 Web UI 交互界面。
diffusers
综合 49
基于 PyTorch 的前沿扩散模型库，支持图像、视频和音频生成。
Real-Time-Voice-Cloning
综合 49
仅需 5 秒即可克隆声音，并实时生成任意语音。
pytorch-image-models
综合 49
PyTorch 图像骨干网络合集，包含各类主流模型、预训练权重及训练推理脚本。
pytorch-lightning
综合 49
无需修改代码，即可在单卡或上万张 GPU 上预训练及微调任意规模的 AI 模型。
tinygrad
综合 47
融合 PyTorch 与 micrograd 理念的极简轻量级深度学习框架。
yolov5
综合 45
基于 PyTorch 的 Ultralytics YOLOv5，支持目标检测、实例分割、分类、模型训练与导出。
pytorch （深度学习框架）
综合 43
支持强大 GPU 加速的 Python 库，可用于张量计算和搭建动态深度神经网络。