pytorch生态项目

pytroch生态 均为Github项目

机器学习

baal (bayesian active learning)

BoTorch(Bayesian Optimization built on PyTorch)

Hummingbird: compiling trained traditional ML models into tensor computations

GPytorch: Gaussian process library implemented using PyTorch

skorch:A scikit-learn compatible neural network library that wraps PyTorch.

pytorchAPI接口

fastai

ignite

pytorch-lightning

Poutyne: Keras-like framework for PyTorch and handles much of the boilerplating code

模型解释性

Captum

训练提速

Horovod:distributed deep learning training framework

glow： machine learning compiler and execution engine for hardware accelerators

深度学习

allennlp

detectron2

flair

MMF:Multi Modal Facebook

optuna：A hyperparameter optimization framework

DGL

catalyst：RL

advertorch

PyTorch-NLP

Pytroch3D: 3D Computer Vision research with PyTorch

Pytorch Geometric: geometric deep learning extension library

NeMO:Neural Modules

可视化

tensorboardX

visdom

Allegro Trains

数据增强

imgaug

albumentations

其他

Crypten:  encrypts and decrypts tensors

kornia:Computer Vision Library for PyTorch

MONAI:Medical Open Network for AI

PySyft:Python library for secure and private Deep Learning