pytorch生态项目
2020-07-13 15:33:11 0 举报AI智能生成
深度学习框架pytorch的生态项目，更高效的深度学习开发
作者其他创作
大纲/内容
机器学习
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
模型解释性
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
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
0 条评论
下一页
