最佳PyTorch工具集合：AI开发者的实用资源
torch-fidelity
PyTorch生成模型评估
PyTorch生成模型评估指标库
Transformer-from-scratch
TransformerLLM
简洁实现Transformer模型的入门教程
pyg-lib
pyg-libPyTorch
图形神经网络高性能计算库
pytorch-image-models
PyTorch图像模型
全面的PyTorch图像模型集合
vision
torchvision计算机视觉
TorchVision 计算机视觉库 提供数据集模型和图像处理功能
sockeye
Sockeye机器翻译
基于PyTorch的开源神经机器翻译工具包
ai-edge-torch
AI Edge TorchPyTorch
PyTorch模型转TensorFlow Lite的开源解决方案
lightning-thunder
PyTorch性能优化
PyTorch模型优化编译器 显著提升训练速度
d2l-pytorch-slides
PyTorch深度学习
基于PyTorch的深度学习教程幻灯片集 涵盖基础到高级主题
extension-cpp
PyTorchC++扩展
PyTorch自定义C++/CUDA运算符扩展示例
FBGEMM
FBGEMM矩阵乘法
针对服务器端推理的高性能低精度矩阵运算库
csprng
PyTorchCSPRNG
PyTorch加密与随机数生成扩展库
data
TorchDataDataLoader
PyTorch数据加载组件的改进版TorchData
cppdocs
PyTorchC++ API
PyTorch C++ API 文档网站的自动生成仓库
captum
Captum模型可解释性
PyTorch模型可解释性和理解的开源库
botorch
BoTorch贝叶斯优化
PyTorch驱动的模块化贝叶斯优化库
builder
PyTorch构建脚本
自动化构建和测试PyTorch二进制文件的工具集
segmentation_models.pytorch
图像分割PyTorch
基于PyTorch的神经网络图像分割库
rcg
RCGPyTorch
RCG框架实现突破性无条件图像生成性能
AttentionDeepMIL
深度学习多实例学习
深度多实例学习的注意力机制算法实现
推荐工具精选
AI云服务特惠
懂AI专属折扣关注微信公众号
最新AI工具、AI资讯
独家AI资源、AI项目落地
微信扫一扫关注公众号