Awesome

Torchpack

Torchpack is a neural network training interface based on PyTorch, with a focus on flexibility.

pip install torchpack