机器学习
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.
Pyro: flexible, scalable deep probabilistic programming library built on PyTorch
pytorchAPI接口
fastai
ignite
pytorch-lightning
Poutyne: Keras-like framework for PyTorch and handles much of the boilerplating code
训练提速
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
其他
Crypten: encrypts and decrypts tensors
ELF: Extensive, Lightweight, and Flexible platform for game research
kornia:Computer Vision Library for PyTorch
MONAI:Medical Open Network for AI
ParlAI:python framework for sharing, training and testing dialogue models
PySyft:Python library for secure and private Deep Learning
