Deep learning with Horovod for distributed training