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

dataloader

dataloader

Merlin Dataloader推荐系统

适用于 TensorFlow、PyTorch 和 JAX 的 GPU 优化数据加载器

malaya

malaya

MalayaPyTorch

马来西亚语自然语言处理库

multidim-positional-encoding

multidim-positional-encoding

Positional EncodingPyTorch

支持PyTorch和TensorFlow的1D、2D和3D正弦位置编码实现

gretel-synthetics

gretel-synthetics

Gretel Synthetics合成数据

提供多模型支持的开源合成数据生成库

Jetson-Nano-Ubuntu-20-image

Jetson-Nano-Ubuntu-20-image

Jetson NanoUbuntu 20.04

Jetson Nano Ubuntu 20.04深度学习环境配置指南

TonY

TonY

TonY深度学习

在 Apache Hadoop 上原生运行深度学习框架的框架

X2Paddle

X2Paddle

深度学习PyTorch

飞桨生态下的模型转换工具,致力于帮助其它深度学习框架用户快速迁移至飞桨框架

elasticdl

elasticdl

ElasticDL深度学习

Kubernetes原生深度学习框架,支持弹性调度和容错

BladeDISC

BladeDISC

BladeDISCPAI-Blade

用于机器学习工作负载的动态形态编译器

breast_cancer_classifier

breast_cancer_classifier

Deep Neural Networksbreast cancer

深度学习模型助力乳腺癌筛查增强放射科医师诊断能力

tfrecord

tfrecord

TFRecordReaderTFRecordWriter

允许在 python 中有效地读取和写入 tfrecord 文件

rl-book

rl-book

Reinforcement LearningTensorFlow

强化学习理论及Python实现的教程和代码

deep-learning-containers

deep-learning-containers

AWS Deep Learning ContainersAmazon SageMaker

高效优化的TensorFlow、PyTorch与MXNet深度学习环境

lambda-packs

lambda-packs

AWS LambdaSelenium PhantomJS

支持各种工具和库的预编译AWS Lambda包

variational-autoencoder

variational-autoencoder

Variational Autoencodertensorflow

变分自编码器参考实现,兼容TensorFlow和PyTorch

ncps

ncps

Neural Circuit PoliciesPyTorch

NCP、LTC 和 CfC 有线神经模型的 PyTorch 和 TensorFlow 实现

torch-mlir

torch-mlir

Torch-MLIRPyTorch

为PyTorch生态系统提供高级编译器支持,并实现与MLIR生态系统的高效集成

TorchSSL

TorchSSL

TorchSSLSemi-supervised learning

优化半监督学习效率与效果的工具箱

fastMRI

fastMRI

fastMRI磁共振成像

原始 MRI 测量值和临床 MRI 图像的大规模数据集

pytorch_tabular

pytorch_tabular

PyTorchPyTorch Lightning

表格数据深度学习的简易化解决方案