Supersedes https://github.com/mlbench/mlbench-core/issues/82. We can now also use PyTorch 1.7.0 - [x] CIFAR10, ResNet20, All Reduce, 1 to 16 workers - [x] CIFAR10, ResNet20, DDP, 1 to 16 workers - [x] Wikitext2, LSTM, All Reduce, 1 to 16 (32 ?) workers - [ ] Wikitext2, LSTM, DDP, 1 to 16 (32 ?) workers - [x] WMT16, LSTM, All Reduce, 1 to 32 workers - [ ] WMT16, LSTM, DDP, 1 to 32 workers - [x] WMT17, Transformer, All Reduce, 1 to 32 workers - [ ] WMT17, Transformer, DDP, 1 to 32 workers