最佳PyTorch工具集合：AI开发者的实用资源
Open3D-ML
Open3D-ML3D机器学习
Open3D 的扩展，用于处理 3D 机器学习任务
koila
KoilaCUDA错误
通过一行代码解决CUDA内存不足错误
adversarial-attacks-pytorch
TorchattacksPyTorch
提供对抗攻击方法的PyTorch库，支持多种攻击技术
LibMTL
LibMTL多任务学习
基于PyTorch的多任务学习开源库，支持多种架构和优化策略
torchrec
TorchRecPyTorch
旨在提供大规模推荐系统所需的常见稀疏性和并行性原语的PyTorch库
hivemind
Hivemind深度学习
去中心化的分布式深度学习PyTorch库
torch-cam
TorchCAMPyTorch
利用 PyTorch 中卷积层的特定于类的激活的简单方法
torchmetrics
TorchMetricsPyTorch
100+ PyTorch 指标实现的集合，以及一个易于使用的 API 来创建自定义指标
neuraloperator
NeuralOperatorPyTorch
用于 在 PyTorch 中学习神经算子的综合性库
nvdiffrec
nvdiffrecNVIDIA
从多视角图像优化3D模型的拓扑结构、材质和光照
webdataset
WebDatasettar文件
大规模深度学习数据集的管理工具
contrastive-unpaired-translation
Contrastive Unpaired TranslationPyTorch
基于对比学习的无监督图像转换
s4
S4HiPPO
多种序列建模模型的官方实现和实验
thorough-pytorch
PyTorch深度学习
从基础到进阶的PyTorch深度学习教程
uvadlc_notebooks
UvA Deep Learning TutorialsPyTorch
深度学习系列教程，覆盖优化、Transformer、图神经网络等多个主题
torchinfo
torchinfoPyTorch
高级模型结构查看工具，适用于PyTorch
torch-points3d
torch-points3d深度学习
用于在点云上进行深度学习的 Pytorch 框架
TensorRT
Torch-TensorRTPyTorch
提升PyTorch推理效率的工具
torchgeo
TorchGeoPyTorch
优化地理空间数据处理的机器学习与遥感工具
openfold
OpenFoldAlphaFold 2
增强蛋白质结构预测功能的AlphaFold2 PyTorch复现版本
推荐工具精选
AI云服务特惠
懂AI专属折扣关注微信公众号
最新AI工具、AI资讯
独家AI资源、AI项目落地
微信扫一扫关注公众号