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