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

benchmark_v0

benchmark_v0

模型PyTorch

基于PytorchModelHubMixin的Hugging Face Hub模型集成

fc_8

fc_8

模型集成PyTorch

PytorchModelHubMixin简化模型部署至Hugging Face Hub

flashtorch

flashtorch

FlashTorchPyTorch

基于PyTorch的神经网络可视化工具

pytorch-generative

pytorch-generative

PyTorch生成模型

PyTorch生成模型开发的全能助手

FLASH-pytorch

FLASH-pytorch

FLASHTransformer

FLASH 线性时间内提升Transformer效能的开源实现

torch-scan

torch-scan

PyTorch模型分析

PyTorch模型分析和性能评估工具

pytracking

pytracking

视觉目标跟踪视频目标分割

基于PyTorch的开源视觉目标跟踪和视频对象分割框架

pytorch-onn

pytorch-onn

PyTorch光子集成电路

基于PyTorch的光子神经网络仿真与优化框架

torchquantum

torchquantum

TorchQuantum量子计算

快速可扩展的PyTorch量子计算框架

EchoTorch

EchoTorch

EchoTorch神经网络

高效回声状态网络研究工具库

docker-pytorch

docker-pytorch

PyTorchDocker

PyTorch开发环境的Docker镜像

mac-ml-speed-test

mac-ml-speed-test

Mac机器学习

Apple Silicon Mac机器学习性能测试工具

torchquad

torchquad

torchquad数值积分

基于GPU加速的开源数值积分框架

mixture-of-experts

mixture-of-experts

PyTorch神经网络

PyTorch实现的稀疏门控专家混合层

optimum-quanto

optimum-quanto

Optimum Quanto量化

PyTorch模型量化框架 提升性能和效率

brevitas

brevitas

Brevitas神经网络量化

面向神经网络量化的PyTorch库

BertBasedCorrectionModels

BertBasedCorrectionModels

BERT文本纠错

基于BERT的中文文本纠错模型集合

recurrent-memory-transformer-pytorch

recurrent-memory-transformer-pytorch

Recurrent Memory TransformerPyTorch

Recurrent Memory Transformer的PyTorch实现助力超长序列处理

spear-tts-pytorch

spear-tts-pytorch

Spear-TTS文本转语音

Pytorch实现的多说话人文本转语音模型

poptorch

poptorch

PopTorchPyTorch

将PyTorch模型部署到Graphcore IPU的开源工具