A library for distributed ML training with PyTorch - View it on GitHub