A high-level model training framework for PyTorch - View it on GitHub