Skip to content

Latest commit

 

History

History
53 lines (22 loc) · 1.2 KB

pytorch.md

File metadata and controls

53 lines (22 loc) · 1.2 KB

Pytorch ecosystem

https://pytorch.org/ecosystem/

Pytorch frameworks

high-level training APIs built on top of pytorch

a comparison article: https://neptune.ai/blog/model-training-libraries-pytorch-ecosystem

https://github.com/yukkyo/Compare-PyTorch-Catalyst-Ignite-Lightning

https://catalyst-team.github.io/catalyst

The lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate

https://pytorch.org/ignite/

  • skorch A scikit-learn compatible neural network library that wraps pytorch

follow the sklearn API

An ML framework to accelerate research and its path to production https://flambe.ai

  • pywick High-level batteries-included neural network training library for Pytorch

  • poutyne A Keras-like framework and utilities for PyTorch