High-level library to help with training neural networks in PyTorch - View it on GitHub