distributed_training_notes DDP code partially adapted from https://github.com/pytorch/examples/tree/main/distributed/ddp-tutorial-series