最佳PyTorch工具集合:AI开发者的实用资源

joeynmt

joeynmt

Joey NMT机器翻译

简洁而清晰的NMT模型实现,促进教育和学习

OpenSTL

OpenSTL

OpenSTL时空预测

OpenSTL:时空预测学习的全面基准和模块化框架

semantic-segmentation

semantic-segmentation

Semantic SegmentationPyTorch

提供丰富数据集和易于定制的语义分割模型

allRank

allRank

allRankPyTorch

深入的PyTorch排序学习框架,支持多种神经网络模型

conformer

conformer

ConformerPyTorch

结合卷积神经网络和Transformers的语音识别模型

former

former

transformerPyTorch

使用PyTorch实现简单Transformer模型的指南

Awesome-Backbones

Awesome-Backbones

Awesome-Backbones图像分类

图像分类的主干网络库及其使用教程

yoloair

yoloair

YOLOv8UltralyticsPro

YOLOAir2024版:综合模型改进教程与源码库

pytorch-seq2seq

pytorch-seq2seq

PyTorchseq2seq

使用PyTorch实现序列到序列模型的教程

mmdetection

mmdetection

OpenMMLabMMDetection

MMDetection:基于PyTorch的高效目标检测工具箱

torchscale

torchscale

TorchScaleTransformers

高效扩展Transformer模型的PyTorch开源库

mmsegmentation

mmsegmentation

MMSegmentationPyTorch

高效的PyTorch语义分割工具箱与新特性介绍

AdaSeq

AdaSeq

AdaSeqModelScope

完善的序列理解模型开发库,涵盖多种高级任务

relora

relora

ReLoRAPyTorch

使用ReLoRA实现高效深度学习模型训练

TinyLLaVA_Factory

TinyLLaVA_Factory

TinyLLaVA Factory多模态模型

模块化的开源小规模多模态模型库

BertWithPretrained

BertWithPretrained

BERTPyTorch

基于PyTorch实现的BERT模型及相关下游任务

keras-nlp

keras-nlp

KerasNLPTensorFlow

兼容多框架的自然语言处理工具和预训练模型

transformers-tutorials

transformers-tutorials

PyTorchTransformers

Transformers模型在自然语言处理中的应用教程

nncf

nncf

Neural Network Compression FrameworkOpenVINO

Neural Network Compression Framework:高效神经网络推理压缩算法

OpenDelta

OpenDelta

OpenDelta参数高效微调

高效参数调整的开源工具包